mindspore.ops.EuclideanNorm

class mindspore.ops.EuclideanNorm(keep_dims=False)[源代码]

计算Tensor维度上元素的欧几里得范数，根据给定的轴对输入进行规约操作。

参数：

  • keep_dims (bool，可选) - 如果为True，被规约的轴保留为1，如果为False，不保留给定的这些轴，默认值：False。

输入：

  • x (Tensor) - 输入Tensor，将被规约，数据类型为：float16、float32、float64、int8、int16、 int32、int64、complex64、complex128、uint8、uint16、uint32、uint64。

  • axes (Tensor) - 将进行规约的轴。数据类型为：int32、int64。取值范围为： [-rank(x), rank(x))。

输出：

Tensor，与输入 x 具有相同的数据类型。

异常：

  • TypeError - 如果 keep_dims 不是一个bool值。

  • TypeError - 如果 x 不是一个Tensor。

  • ValueError - 如果 axes 超出取值范围。

支持平台：

GPU

样例：

>>> x = Tensor(np.array([[3, 5], [4, 12]])).astype(np.int32)
>>> axes = Tensor([0])
>>> op = ops.EuclideanNorm(keep_dims=True)
>>> output = op(x, axes)
>>> print(output)
[[5 13]]