Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Commit

Permalink
add github link for issue with reshape
Browse files Browse the repository at this point in the history
  • Loading branch information
vandanavk committed Dec 28, 2018
1 parent ebbe347 commit 49ab527
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions python/mxnet/contrib/onnx/onnx2mx/_op_translations.py
Original file line number Diff line number Diff line change
Expand Up @@ -732,6 +732,7 @@ def hardmax(attrs, inputs, proto_obj):
# since reshape doesn't take a tensor for shape,
# computing with np.prod. This needs to be changed to
# to use mx.sym.prod() when mx.sym.reshape() is fixed.
# (https://github.com/apache/incubator-mxnet/issues/10789)
new_shape = (int(np.prod(input_shape[:axis])),
int(np.prod(input_shape[axis:])))
reshape_op = symbol.reshape(inputs[0], new_shape)
Expand Down

0 comments on commit 49ab527

Please sign in to comment.