mindspore.ops.L2Loss

查看源文件
class mindspore.ops.L2Loss[源代码]

用于计算L2范数的一半,但不对结果进行开方操作。

把输入设为x,输出设为loss。

\[loss = \frac{\sum x ^ 2}{2}\]
输入:
  • input_x (Tensor) - 用于计算L2范数的Tensor。数据类型必须为float16或float32。

输出:

Tensor,具有与 input_x 相同的数据类型的Scalar Tensor。

异常:
  • TypeError - input_x 不是Tensor。

  • TypeError - input_x 的数据类型既不是float16也不是float32。

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore
>>> import numpy as np
>>> from mindspore import Tensor, ops
>>> input_x = Tensor(np.array([1, 2, 3]), mindspore.float16)
>>> l2_loss = ops.L2Loss()
>>> output = l2_loss(input_x)
>>> print(output)
7.0