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