乘积

x=torch.tensor([1,2,3])
y=torch.tensor([1,2,3])

  1. Hadamard乘积:x*y 即元素对应相乘
  2. 点积:torch.dot(x,y) 相当于 torch.sum(x*y) 所有元素乘积的和
  3. 矩阵与向量乘积:A矩阵,x向量,Ax就是A中各行与x相乘得到一个元素,结果形状是A的行数。torch.mv(A,x)
  4. 矩阵之间乘积:torch.mm(A,B)

加法降维

  1. x.sum() 实现对x所有元素的加和
  2. x.sum(axis=0,keepdims=True) 实现按列加和,得到一列;axis=1按行求和得到一行,keepdims保持原有维度
  3. A.cumsum(axis=0) 累积求和,从第一行开始,每行加上上一行得到结果

范数求解

torch.norm(x,p=,dim=) 对x 求p(设置范数种类int, float, inf, -inf, ‘fro’, ‘nuc’, optional)范数,并且对某一制定维度做范数,默认为2范数,异常值对1范数的影响较小。

更多推荐