mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

fix default value of axis of argmin

This commit is contained in:
liangcanli 2023-01-30 09:32:16 +08:00
parent 63a8b36505
commit 4a189cd1a9
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/python/mindspore/ops/function/math_func.py
View File

@ -436,7 +436,7 @@ def exp2(x):
return exp2_(tensor_2, x)
def argmin(x, axis=-1, keepdims=False):
def argmin(x, axis=None, keepdims=False):
"""
Returns the indices of the minimum value of a tensor across the axis.