Merge pull request #1 from vlavla/mxnet1.5.0-fix

Mxnet1.5.0 fix
HawkAaron · Nov 1, 2019 · 869e473 · 869e473
2 parents 53cef43 + b7c8646
commit 869e473
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/rnnt_loss-inl.h b/rnnt_loss-inl.h
@@ -227,7 +227,7 @@ class RNNTLossProp : public OperatorProperty {
     CHECK_EQ(llshape[0], lshape[0])
         << "The batch size for the labels and label lengths must be the same.";
 
-    TShape oshape(1);
+    TShape oshape(1, 1);
     oshape[0] = dshape[0];  // batch size
     out_shape->clear();
     out_shape->push_back(oshape);