bugfix flat23d

stefanhenneking · Jun 17, 2016 · f57e292 · f57e292
1 parent f5c6a2e
commit f57e292
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;