Функция torch.renorm в PyTorch

Pytorch torch 参考手册Справочное руководство Pytorch torch

torch.renormЭто функция в PyTorch, используемая для перенормировки тензоров. Она нормализует тензор в соответствии с указанной нормой, так, чтобы норма элементов вдоль указанного измерения не превышала заданное значение.

Определение функции

torch.renorm(input, p, dim, maxnorm)

Описание параметров:

  • input: Входной тензор
  • p: Порядок нормы
  • dim: Размерность для нормализации
  • maxnorm: Максимальное значение нормы

Пример использования

Пример

import torch

# Создать тензор
x = torch.tensor([[2.0, 4.0, 6.0], [3.0, 6.0, 9.0]])

# Выполнить нормализацию по L2-норме по dim=1 с максимальной нормой 1
y = torch.renorm(x, p=2, dim=1, maxnorm=1)
print(y)

Результат вывода:

tensor([[0.2673, 0.5345, 0.8018],
        [0.2673, 0.5345, 0.8018]])

Пример

import torch

# Создать тензор
x = torch.tensor([[1.0, 2.0, 3.0]])

# Выполнить нормализацию по L1-норме по dim=1
y = torch.renorm(x, p=1, dim=1, maxnorm=1)
print(y)

Результат вывода:

tensor([[0.1667, 0.3333, 0.5000]])

Pytorch torch 参考手册Справочное руководство Pytorch torch

Другие расширения