tensor的统计属性
常见的统计属性:norm(范数),mean(均值), sum(求和),prod(累乘),max(最大值),min(最小值),argmin(最小值位置),argmax(最大值位置),kthvalue(求第k个小的值的位置),topk(求前k个大的值)
norm
norm-p
.norm(1)一范数,所有元素绝对值求和
.norm(2)二范数,所有元素绝对值的平方和,再开根
.norm(n,dim=d)指定维度上的范数,取哪个维度那个维度会被消掉。
import torch
a=torch.full([8],1)
b=a.view(2,4)
c=a.view(2,2,2)
print(b)
print(c)
print(a.norm(1),b.norm(1),c.norm(1))
print(a.norm(2),b.norm(2),c.norm(2))
print(b.norm(1,dim=1))
print(b.norm(2,dim=1))
print(c.norm(1,dim=1))
print(c.norm(2,dim=1))