Fix crashing softmax mxnet backend operator on GPUs (#107)

awslabs · May 27, 2018 · 546b50f · 546b50f
1 parent 4857329
commit 546b50f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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