diff --git a/mindspore/ops/operations/nn_ops.py b/mindspore/ops/operations/nn_ops.py index 6823e06a79..7ec333c96d 100644 --- a/mindspore/ops/operations/nn_ops.py +++ b/mindspore/ops/operations/nn_ops.py @@ -2339,7 +2339,17 @@ class SparseSoftmaxCrossEntropyWithLogits(PrimitiveWithInfer): ``GPU`` ``CPU`` Examples: - Please refer to :class:`mindspore.ops.SparseSoftmaxCrossEntropyWithLogits` + >>> logits = Tensor([[2, 4, 1, 4, 5], [2, 1, 2, 4, 3]], mindspore.float32) + >>> labels = Tensor([0, 1], mindspore.int32) + >>> sparse_softmax_cross = ops.SparseSoftmaxCrossEntropyWithLogits() + >>> loss = sparse_softmax_cross(logits, labels) + >>> print(loss) + 3.4878292 + >>> sparse_softmax_cross_grad = ops.SparseSoftmaxCrossEntropyWithLogits(is_grad=True) + >>> loss_grad = sparse_softmax_cross_grad(logits, labels) + >>> print(loss_grad) + [[-0.48415753 0.04306427 0.00582811 0.11706084 0.3182043 ] + [ 0.04007946 -0.4852556 0.04007946 0.2961494 0.10894729]] """ @prim_attr_register