torch.renorm

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

Возвращает тензор, где каждый подтензор input вдоль размерности dim нормализован таким образом, что 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]])