torch.mul函数使用需要注意的点

torch.mul函数使用需要注意的点

用的时候，可不能搞混了！

---

• torch.mul(a,b)是a和b对应元素相乘
• np.matmul(a,b)是矩阵相乘，即行乘列
• torch.matmul(a,b)是矩阵相乘，即行乘列

  ---

  发现个有用的，tensor类型转换：

• torch._cast_Float（input）
  方便！