diff --git a/mindspore/ops/operations/array_ops.py b/mindspore/ops/operations/array_ops.py
index a7c3f50440..0af284b947 100644
--- a/mindspore/ops/operations/array_ops.py
+++ b/mindspore/ops/operations/array_ops.py
@@ -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
diff --git a/mindspore/ops/operations/nn_ops.py b/mindspore/ops/operations/nn_ops.py
index f5037882f1..42ae394289 100644
--- a/mindspore/ops/operations/nn_ops.py
+++ b/mindspore/ops/operations/nn_ops.py
@@ -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),