[Fix test backend] Fix softmax in mxnet backend (#828)

3386f466 · VoVAllen · GitHub · 87310503 · 3386f466
Unverified Commit 3386f466 authored Sep 04, 2019 by VoVAllen Committed by GitHub Sep 04, 2019
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tests/backend/mxnet/__init__.py tests/backend/mxnet/__init__.py +1 -1

No files found.
--- a/tests/backend/mxnet/__init__.py
+++ b/tests/backend/mxnet/__init__.py
@@ -54,7 +54,7 @@ def reduce_sum(x):
    return x.sum()

 def softmax(x, dim):
-    return nd.softmax(x, dim)
+    return nd.softmax(x, axis=dim)

 def spmm(x, y):
    return nd.dot(x, y)