Merge pull request apache#107 from tqchen/master

Add ProdShape
stefanhenneking · Apr 11, 2016 · 68d900f · 68d900f
2 parents 9ff5a2f + 202d06f
commit 68d900f
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/mshadow/tensor_blob.h b/mshadow/tensor_blob.h
@@ -188,6 +188,19 @@ struct TShape {
     s.shape_[0] = ymax;
     return s;
   }
+  /*!
+   * \return product shape in [dimstart,dimend)
+   * \param dimstart start dimension
+   * \param dimend end dimension
+   */
+  inline index_t ProdShape(int dimstart, int dimend) const {
+    index_t num = 1;
+    const index_t *d = this->data();
+    for (int i = dimstart; i < dimend; ++i) {
+      num *= d[i];
+    }
+    return num;
+  }
   /*!
    * \brief get the shape of tensor specifying dim
    * \return the shape requested