关于范数的定义,有多种情况,如下所示:

0 范数:当前向量中的元素之和
1 范数:对应向量元素只差的绝对值之和
2 范数: 元素的平方和再开方
p 范数: 元素的绝对值的P次方之和再开P次方

import torch

a = torch.rand(2, 1)

# a 的一范数
print(torch.norm(a, p=1))
# a 的二范数
print(torch.norm(a))