torch.norm
函数用于计算张量的范数(norm),可以理解为张量的“长度”或“大小”。根据范数的不同类型,它可以衡量不同的张量性质。该函数可以计算 向量 和 矩阵 的多种范数,如 L1范数、L2范数、无穷范数 等。
1. 函数签名
torch.norm(input, p='fro', dim=None, keepdim=False, dtype=None, out=None)
-
input: 需要计算范数的输入张量。
-
p: 范数的类型,常用值包括:
'fro'
(默认值):Frobenius 范数(矩阵的元素平方和开平方,类似于 L2 范数)。p=1
:L1 范数,元素的绝对值和。p=2
:L2 范数,元素的平方和的平方根(也称为欧几里得范数)。p=float('inf')
:无穷范数,张量元素的最大绝对值。- 其他 p 值可以表示不同的 p-范数(如 p=3 表示元素的三次方和的开三次方根)。
-
dim: 计算范数的维度。如