提交 22eeabdc 编写于 作者: L liuwenhao4

Fix some mistakes of ArgMaxWithValue vm ops

上级 af2ebca0
...@@ -1123,7 +1123,7 @@ class ArgMaxWithValue(PrimitiveWithInfer): ...@@ -1123,7 +1123,7 @@ class ArgMaxWithValue(PrimitiveWithInfer):
- output_x (Tensor) - The maximum value of input tensor, the shape same as index. - output_x (Tensor) - The maximum value of input tensor, the shape same as index.
Examples: Examples:
>>> input_x = Tensor(np.random.rand(5)) >>> input_x = Tensor(np.random.rand(5), mindspore.float32)
>>> index, output = P.ArgMaxWithValue()(input_x) >>> index, output = P.ArgMaxWithValue()(input_x)
""" """
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册