Функция torch.norm в PyTorch
Справочное руководство PyTorch torch
torch.normЭто функция в PyTorch, используемая для возврата нормы тензора. По умолчанию она вычисляет норму L2 (евклидову норму).
Определение функции
torch.norm(input, p='fro', dim, keepdim=False)
Пример использования
Пример
import torch
x = torch.tensor([3.0, 4.0])
# Норма L2 (по умолчанию)
print("Норма L2:", torch.norm(x))
# Норма L1
print("Норма L1:", torch.norm(x, p=1))
# Норма L2 (с использованием dim)
y = torch.tensor([[1.0, 2.0], [3.0, 4.0]])
print("Норма L2 матрицы:", torch.norm(y))
print("Норма L2 по dim=0:", torch.norm(y, dim=0))
print("Норма L2 по dim=1:", torch.norm(y, dim=1))
x = torch.tensor([3.0, 4.0])
# Норма L2 (по умолчанию)
print("Норма L2:", torch.norm(x))
# Норма L1
print("Норма L1:", torch.norm(x, p=1))
# Норма L2 (с использованием dim)
y = torch.tensor([[1.0, 2.0], [3.0, 4.0]])
print("Норма L2 матрицы:", torch.norm(y))
print("Норма L2 по dim=0:", torch.norm(y, dim=0))
print("Норма L2 по dim=1:", torch.norm(y, dim=1))
Вывод:
L2 范数: tensor(5.) L1 范数: tensor(7.) 矩阵 L2 范数: tensor(5.477) 按 dim=0 L2 范数: tensor([3.162, 4.472]) 按 dim=1 L2 范数: tensor([2.236, 5.000])
Другие расширения