torch.mul()、torch.mm()、torch.dot()和torch.mv()之间的区别

  1. torch.mul()是矩阵的点乘,即对应的位相乘,要求shape一样, 返回的还是个矩阵
  2. torch.mm()是矩阵正常的矩阵相乘,(a, b)* ( b, c ) = ( a, c )
  3. torch.dot()类似于mul(),它是向量(即只能是一维的张量)的对应位相乘再求和,返回一个tensor数值
  4. torch.mv()是矩阵和向量相乘,类似于torch.mm()
import torch
A = torch.tensor([[1,2,3],
                   [4,5,6]])
x = torch.tensor([1,2,3])
torch.mv(A, x) # 第一个参数是矩阵,第二个参数只能是一维向量 

输出:tensor([14, 32])

等价于A x T x^T xT

你可能感兴趣的:(pytorch,pytorch)