1.加法 按照维度运算
按照维度运算结果是如何得到的?
参考理解PyTorch中的dimensions维度,三维张量求和过程-CSDN博客
保留维度 以便广播
keepdim=True
2.累加
a=torch.arange(40).reshape((2,5,4))
print(a)
print()
print(a.sum(axis=[0]))
print()
print(a.cumsum(axis=0))
累加为按照所选维度一层整体加到下一层。
3.向量点积
内积
4.矩阵向量积 mv
a=torch.arange(12).reshape((3,4))
b=torch.ones(12).reshape(4,3)
c=torch.ones(4,dtype=int)
a,torch.mv(a,c)
其中,c为列向量,a,c前后顺序也很重要
5.矩阵乘法 mm
6.范数
对向量:
对矩阵而言
解答
1.copy不复制内存 clone占内存
2.torch 向量可以横看也可以竖看
行向量和列向量只存在于二维及以上维度数组中