p60975123
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!391 Modify examples of ApplyMomentum/ArgMax 

Merge pull request !391 from Simson/push-to-opensource
This commit is contained in:
mindspore-ci-bot 2020-04-16 20:41:27 +08:00 committed by Gitee
parent d3e4dd06b0 51de42bc05
commit 82cfe347fb
2 changed files with 2 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mindspore/ops/operations/array_ops.py
View File

@ -983,8 +983,7 @@ class Argmax(PrimitiveWithInfer):
Examples:
>>> input_x = Tensor(np.array([2.0, 3.1, 1.2]))
>>> index = P.Argmax()(input_x)
>>> assert index == Tensor(1, mindspore.int64)
>>> index = P.Argmax(output_type=mindspore.int32)(input_x)
"""
@prim_attr_register

6
mindspore/ops/operations/nn_ops.py
View File

@ -1345,11 +1345,7 @@ class ApplyMomentum(PrimitiveWithInfer):
Tensor, parameters to be updated.
Examples:
>>> net = ResNet50()
>>> loss = nn.SoftmaxCrossEntropyWithLogits()
>>> opt = P.ApplyMomentum(Tensor(np.array([0.001])), Tensor(np.array([0.9])),
filter(lambda x: x.requires_grad, net.get_parameters()))
>>> model = Model(net, loss, opt)
Please refer to the usage in nn.ApplyMomentum.
"""
__mindspore_signature__ = (
('variable', sig_rw.RW_WRITE, sig_kind.KIND_POSITIONAL_KEYWORD),