Browse Source

!15047 update redirect of sparsesoftmaxcrossentropywithlogits

From: @mind-lh
Reviewed-by: @liangchenghui,@wuxuejian
Signed-off-by: @liangchenghui
pull/15047/MERGE
mindspore-ci-bot Gitee 4 years ago
parent
commit
5c77705488
1 changed files with 11 additions and 1 deletions
  1. +11
    -1
      mindspore/ops/operations/nn_ops.py

+ 11
- 1
mindspore/ops/operations/nn_ops.py View File

@@ -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


Loading…
Cancel
Save