Merge pull request apache#133 from tqchen/master

bugfix flat23d
stefanhenneking · Jun 17, 2016 · 02a4a0e · 02a4a0e
2 parents 6613f62 + f57e292
commit 02a4a0e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mshadow/tensor_blob.h b/mshadow/tensor_blob.h
@@ -221,7 +221,7 @@ struct TShape {
     for (index_t i = 0; i < axis; ++i) {
       s.shape_[0] *= d[i];
     }
-    for (index_t i = axis; i < ndim_; ++i) {
+    for (index_t i = axis + 1; i < ndim_; ++i) {
       s.shape_[2] *= d[i];
     }
     return s;