torch.renorm

torch.renorm(input, p, dim, maxnorm, *, out=None) → 張量

返回一個張量，其中沿維度 dim 的 input 的每個子張量都經過歸一化，使得該子張量的 p-範數小於 maxnorm 的值

注意

如果一個子張量的範數低於 maxnorm，則該子張量保持不變

引數

• input (張量) – 輸入張量。

• p (浮點數) – 範數計算中的冪

• dim (整數) – 用於切片獲取子張量的維度

• maxnorm (浮點數) – 使每個子張量保持在最大範數之下

關鍵字引數

out (張量, 可選) – 輸出張量。

示例

>>> x = torch.ones(3, 3)
>>> x[1].fill_(2)
tensor([ 2.,  2.,  2.])
>>> x[2].fill_(3)
tensor([ 3.,  3.,  3.])
>>> x
tensor([[ 1.,  1.,  1.],
        [ 2.,  2.,  2.],
        [ 3.,  3.,  3.]])
>>> torch.renorm(x, 1, 0, 5)
tensor([[ 1.0000,  1.0000,  1.0000],
        [ 1.6667,  1.6667,  1.6667],
        [ 1.6667,  1.6667,  1.6667]])