diff --git a/keras/backend/mxnet_backend.py b/keras/backend/mxnet_backend.py index 7af184991de..b1137f31319 100644 --- a/keras/backend/mxnet_backend.py +++ b/keras/backend/mxnet_backend.py @@ -2809,7 +2809,7 @@ def softmax(x): # Returns A tensor. """ - return KerasSymbol(mx.sym.SoftmaxActivation(data=x.symbol)) + return KerasSymbol(mx.sym.softmax(data=x.symbol)) @keras_mxnet_symbol