torch.linalg.norm函数

torch.linalg.norm函数是PyTorch中用于计算张量或向量的范数的函数。它的语法如下：

torch.linalg.norm(input, ord=None, dim=None, keepdim=False, out=None, dtype=None)

其中，参数input表示输入的张量或向量，ord表示计算的范数类型，dim表示在哪个维度上计算范数，keepdim表示是否保持计算后的维度数与输入的维度数相同，out表示输出张量，dtype表示输出张量的数据类型。

常见的范数类型有：

- ord=1表示计算L1范数
- ord=2表示计算L2范数
- ord=float('inf')表示计算无穷范数

例如，计算张量x在第1维上的L2范数：

import torch

x = torch.randn(3, 4, 5)
norm = torch.linalg.norm(x, ord=2, dim=1, keepdim=True)
print(norm.shape)  # 输出torch.Size([3, 1, 5])