mindelec.loss.get_loss_metric
- mindelec.loss.get_loss_metric(name)[源代码]
获取损失函数。
- 参数:
name (str) - 损失函数的名称。
- 返回:
Function,损失函数。
- 支持平台:
Ascend
样例:
>>> import numpy as np >>> from mindelec.loss import get_loss_metric >>> import mindspore >>> from mindspore import Tensor >>> l1_loss = get_loss_metric('l1_loss') >>> logits = Tensor(np.array([1, 2, 3]), mindspore.float32) >>> labels = Tensor(np.array([[1, 1, 1], [1, 2, 2]]), mindspore.float32) >>> output = l1_loss(logits, labels) >>> print(output) 0.6666667