change argument order (#20413)

apache · Jul 2, 2021 · 0716dc0 · 0716dc0
1 parent 2a1f356
commit 0716dc0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/mxnet/numpy_extension/_op.py b/python/mxnet/numpy_extension/_op.py
@@ -30,7 +30,7 @@
 
 # pylint: disable=too-many-arguments
 @set_module('mxnet.numpy_extension')
-def softmax(data, axis=-1, length=None, temperature=None, use_length=False, dtype=None):
+def softmax(data, length=None, axis=-1, temperature=None, use_length=False, dtype=None):
     r"""Applies the softmax function.
 
     The resulting array contains elements in the range (0,1) and the elements along the given axis sum up to 1.