torch.mul(a, b)用法

• torch.mul作element-wise的矩陣點乘，維數不限，可以矩陣乘標量
• 點乘都是broadcast的，可以用 torch.mul(a, b) 實現，也可以直接用 * 實現。
• 當a, b維度不一致時，會自動填充到相同維度相點乘。

  import torch
  
  a = torch.ones(3,4)
  print(a)
  b = torch.Tensor([1,2,3]).reshape((3,1))
  print(b)
  
  print(torch.mul(a, b))

  tensor([[1., 1., 1., 1.],
          [1., 1., 1., 1.],
          [1., 1., 1., 1.]])
  tensor([[1.],
          [2.],
          [3.]])
  tensor([[1., 1., 1., 1.],
          [2., 2., 2., 2.],
          [3., 3., 3., 3.]])

  注：來自pytorch函數torch中的一些乘法和點乘法焊槍.mm（一）矩陣乘法火炬.mul()， 火炬.matmul高維張量乘維數要求,之,幾種,torchmm,torchmultorchmatmul,Tensor,相乘,維度 (pythonf.cn)