mindspore.nn.HingeEmbeddingLoss
- class mindspore.nn.HingeEmbeddingLoss(margin=1.0, reduction='mean')[source]
Calculate the Hinge Embedding Loss value based on the input ‘logits’ and’ labels’ (only including 1 or -1). Usually used to measure the similarity between two inputs.
The loss function for
-th sample in the mini-batch isand the total loss functions is
where
.- Parameters
- Inputs:
logits (Tensor) - The predicted value, expressed as
in the equation. Tensor of shape where means any number of dimensions.labels (Tensor) - Label value, represented as
in the equation. Same shape as the logits, contains -1 or 1.
- Returns
Tensor or Tensor scalar, the computed loss depending on
.- Raises
TypeError – If logits is not a Tensor.
TypeError – If labels is not a Tensor.
TypeError – If margin is not a float or int.
ValueError – If labels does not have the same shape as logits or they could not broadcast to each other.
ValueError – If reduction is not one of ‘none’, ‘mean’, ‘sum’.
- Supported Platforms:
Ascend
GPU
CPU
Examples
>>> import numpy as np >>> from mindspore import Tensor >>> import mindspore.nn as nn >>> import mindspore.common.dtype as mstype >>> arr1 = np.array([0.9, -1.2, 2, 0.8, 3.9, 2, 1, 0, -1]).reshape((3, 3)) >>> arr2 = np.array([1, 1, -1, 1, -1, 1, -1, 1, 1]).reshape((3, 3)) >>> logits = Tensor(arr1, mstype.float32) >>> labels = Tensor(arr2, mstype.float32) >>> loss = nn.HingeEmbeddingLoss(reduction='mean') >>> output = loss(logits, labels) >>> print(output) 0.16666667