torch.renorm

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

Возвращает тензор, в котором каждый подтензор input вдоль измерения dim нормализован так, что норма p подтензора меньше значения maxnorm

Примечание

Если норма строки меньше maxnorm, строка не изменяется

Параметры:

• input (Tensor) – входной тензор.
• p (float) – степень для вычисления нормы
• dim (int) – измерение, вдоль которого выполняется разбиение на подтензоры
• maxnorm (float) – максимальная норма каждого подтензора

Именованные аргументы:

out (Tensor, необязательно) – выходной тензор.

Пример:

>>> 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]])