From 4a189cd1a9ec7c812fd9579766dc108c78739caa Mon Sep 17 00:00:00 2001 From: liangcanli Date: Mon, 30 Jan 2023 09:32:16 +0800 Subject: [PATCH] fix default value of axis of argmin --- mindspore/python/mindspore/ops/function/math_func.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/python/mindspore/ops/function/math_func.py b/mindspore/python/mindspore/ops/function/math_func.py index 59cb80276e4b..ce001da409b6 100644 --- a/mindspore/python/mindspore/ops/function/math_func.py +++ b/mindspore/python/mindspore/ops/function/math_func.py @@ -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. -- Gitee