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

Commit

Permalink
fix onnx2mx
Browse files Browse the repository at this point in the history
  • Loading branch information
Wei Chu committed Dec 18, 2020
1 parent 0158a5a commit f866fd0
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion python/mxnet/contrib/onnx/onnx2mx/_op_translations.py
Original file line number Diff line number Diff line change
Expand Up @@ -314,7 +314,7 @@ def _selu(attrs, inputs, proto_obj):
def softmax(attrs, inputs, proto_obj):
"""Softmax function."""
if 'axis' not in attrs:
attrs = translation_utils._add_extra_attributes(attrs, {'axis': 1})
attrs = translation_utils._add_extra_attributes(attrs, {'axis': -1})
return 'softmax', attrs, inputs

def log_softmax(attrs, inputs, proto_obj):
Expand Down

0 comments on commit f866fd0

Please sign in to comment.