From 22eeabdc5f37fe239993fbc590f66749dda2cedb Mon Sep 17 00:00:00 2001 From: liuwenhao4 Date: Tue, 23 Jun 2020 22:53:45 +0800 Subject: [PATCH] Fix some mistakes of ArgMaxWithValue vm ops --- mindspore/ops/operations/array_ops.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/ops/operations/array_ops.py b/mindspore/ops/operations/array_ops.py index 1bb39d1547..f59a3a37e5 100644 --- a/mindspore/ops/operations/array_ops.py +++ b/mindspore/ops/operations/array_ops.py @@ -1123,7 +1123,7 @@ class ArgMaxWithValue(PrimitiveWithInfer): - output_x (Tensor) - The maximum value of input tensor, the shape same as index. Examples: - >>> input_x = Tensor(np.random.rand(5)) + >>> input_x = Tensor(np.random.rand(5), mindspore.float32) >>> index, output = P.ArgMaxWithValue()(input_x) """