Función torch.norm de PyTorch

Pytorch torch 参考手册 Manual de referencia de Pytorch torch

torch.normEs una función en PyTorch que se utiliza para devolver la norma de un tensor. Por defecto calcula la norma L2 (norma euclidiana).

Definición de la función

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

Ejemplo de uso

Ejemplo

import torch

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

# Norma L2 (predeterminada)
print("Norma L2:", torch.norm(x))

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

# Norma L2 (usando dim)
y = torch.tensor([[1.0, 2.0], [3.0, 4.0]])
print("Norma L2 de la matriz:", torch.norm(y))
print("Norma L2 según dim=0:", torch.norm(y, dim=0))
print("Norma L2 según dim=1:", torch.norm(y, dim=1))

El resultado de salida es:

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 参考手册 Manual de referencia de Pytorch torch

Otras extensiones