mindspore.ops.SmoothL1Loss
- class mindspore.ops.SmoothL1Loss(beta=1.0, reduction='none')[source]
Calculate the smooth L1 loss, and the L1 loss function has robustness.
Refer to
mindspore.ops.smooth_l1_loss()
for more details.- Parameters
- Inputs:
logits (Tensor) - Input Tensor of any dimension. Data type must be float16, float32 or float64.
labels (Tensor) - Ground truth data, has the same shape and dtype as the logits.
- Outputs:
Tensor, loss float tensor, same shape and dtype as the logits.
- Supported Platforms:
Ascend
GPU
CPU
Examples
>>> import mindspore >>> import numpy as np >>> from mindspore import Tensor, ops >>> loss = ops.SmoothL1Loss() >>> logits = Tensor(np.array([1, 2, 3]), mindspore.float32) >>> labels = Tensor(np.array([1, 2, 2]), mindspore.float32) >>> output = loss(logits, labels) >>> print(output) [0. 0. 0.5]