mindflow.loss.get_loss_metric

mindflow.loss.get_loss_metric(name)[源代码]

获取损失函数。

参数:
  • name (str) - 损失函数的名称。

返回:

Function,损失函数。

支持平台:

Ascend GPU

样例:

>>> import numpy as np
>>> from mindflow.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