From a35d3441d48f6acfc78ef9d48f182367725cb9b1 Mon Sep 17 00:00:00 2001 From: Jonathan Yan Date: Wed, 11 Nov 2020 13:22:53 -0500 Subject: [PATCH] SampledSoftmaxLoss example --- mindspore/nn/loss/loss.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/mindspore/nn/loss/loss.py b/mindspore/nn/loss/loss.py index 63cd5c1530..2904b697f6 100644 --- a/mindspore/nn/loss/loss.py +++ b/mindspore/nn/loss/loss.py @@ -291,6 +291,15 @@ class SampledSoftmaxLoss(_Loss): Outputs: Tensor, a tensor of shape (N) with the per-example sampled softmax losses. + Examples: + >>> loss = nn.SampledSoftmaxLoss(num_sampled=4, num_classes=7, num_true=1) + >>> weights = Tensor(np.random.randint(0, 9, [7, 10]), mindspore.float32) + >>> biases = Tensor(np.random.randint(0, 9, [7]), mindspore.float32) + >>> labels = Tensor([0, 1, 2]) + >>> inputs = Tensor(np.random.randint(0, 9, [3, 10]), mindspore.float32) + >>> output = loss(weights, biases, labels, inputs) + >>> print(output) # output is ranndom + [ 4.0181947 46.050743 7.0009117] """ def __init__(self, num_sampled, num_classes, num_true=1,