Fonction torch.norm de PyTorch

Pytorch torch 参考手册 Manuel de référence PyTorch torch

torch.normIl s'agit d'une fonction de PyTorch utilisée pour renvoyer la norme d'un tenseur. Par défaut, elle calcule la norme L2 (norme euclidienne).

Définition de fonction

torch.norm(input, p='fro', dim, keepdim=False)

Exemple d'utilisation

Exemple

import torch

x = torch.tensor([3.0, 4.0])

# Norme L2 (par défaut)
print("Norme L2 :", torch.norm(x))

# Norme L1
print("Norme L1 :", torch.norm(x, p=1))

# Norme L2 (en utilisant dim)
y = torch.tensor([[1.0, 2.0], [3.0, 4.0]])
print("Norme L2 de la matrice :", torch.norm(y))
print("Norme L2 selon dim=0 :", torch.norm(y, dim=0))
print("Norme L2 selon dim=1 :", torch.norm(y, dim=1))

La sortie est :

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

Pytorch torch 参考手册 Manuel de référence PyTorch torch

Autres extensions