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)