torch.renorm

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

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

Примечание

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

Параметры:

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

Ключевые аргументы:

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