Pytorch基础 - 2. torch.linalg.norm() 和 torch.linalg.vector_norm() 和 torch.linalg.matrix_norm()

目录

1. torch.linalg.norm() 

2. torch.linalg.vector_norm()

3. torch.linalg.matrix_norm()

---

最近在写程序的时候，用到了L2范数，也因此了解到了这几个函数torch.norm()【已弃用】，torch.linalg.norm()，torch.linalg.matrix_norm()，torch.linalg.vector_norm()。

1. torch.linalg.norm() 

 该函数能够计算向量/矩阵/范数，首先献上官方文档：torch.linalg.norm — PyTorch 1.13

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

(1) ord是指定的范数，默认None，为F-范数(2-范数)。当p='nuc'时为核范数，剩下的可看文档。

(2) dim为指定求范数的维度

• 如果dim是一个int，将计算向量范数。
• 如果dim是一个二元组，将计算矩阵范数。
• 如果dim= None和ord= None，则A将被展为1D，并计算结果向量的2范数。
• 如果dim= None和ord !