Skip to content

GitLab

Unverified Commit 3386f466 authored by VoVAllen's avatar VoVAllen Committed by GitHub
Browse files

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

parent 87310503
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/backend/mxnet/__init__.py
View file @ 3386f466
...@@ -54,7 +54,7 @@ def reduce_sum(x): ...@@ -54,7 +54,7 @@ def reduce_sum(x):
return x.sum() return x.sum()
def softmax(x, dim): def softmax(x, dim):
return nd.softmax(x, dim) return nd.softmax(x, axis=dim)
def spmm(x, y): def spmm(x, y):
return nd.dot(x, y) return nd.dot(x, y)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment