mindspore.ops.renorm
- mindspore.ops.renorm(input_x, p, dim, maxnorm)[源代码]
沿维度 dim 重新规范输入 input_x 的子张量,并且每个子张量的p范数不超过给定的最大范数 maxnorm 。 如果子张量的p范数小于 maxnorm ,则当前子张量不需要修改;否则该子张量需要修改为对应位置的原值除以该子张量的p范数,然后再乘上 maxnorm 。
参数:
input_x (Tensor) - 输入张量。
p (int) - 范数计算的幂。
dim (int) - 获得子张量的维度。
maxnorm (float32) - 给定的最大范数。
返回:
Tensor,shape和type与输入Tensor一致。
异常:
TypeError - p 不是int类型。
TypeError - dim 不是int类型。
TypeError - maxnorm 不是float32类型。
ValueError - p 小于等于0。
- 支持平台:
Ascend
CPU
GPU
样例:
>>> x = Tensor(np.array([[1, 1, 1], [2, 2, 2], [3, 3, 3]]), mindspore.float32) >>> y = ops.renorm(x, p=1, dim=0, maxnorm=5.) >>> print(y) [[1. 1. 1. ] [1.6666666 1.6666666 1.6666666 ] [1.6666667 1.6666667 1.6666667 ]]