Skip to content

M
mindspore

magicwindyyd / mindspore
与 Fork 源项目一致

Fork自 MindSpore / mindspore

通知 1
Star 1
Fork 0
M
mindspore
提交 22eeabdc 编写于 作者: L liuwenhao4
浏览文件
操作

Fix some mistakes of ArgMaxWithValue vm ops

上级 af2ebca0
隐藏空白更改
内联 并排
Showing with 1 addition and 1 deletion
mindspore/ops/operations/array_ops.py
浏览文件 @ 22eeabdc
...@@ -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.
先完成此消息的编辑！
想要评论请 注册 或