pytorch 记录
torch.take() 把张量化为一个行向量,取其中的元素
广播机制中0维和1维的轴是可广播的
矩阵相乘
*和torch.mul()是对应位置相乘
@和torch.mm()是矩阵相乘 torch.bmm()相比mm增加了batch运算但不能广播
torch.matmul()相比bmm更是增加了广播,泛用性强
*和torch.mul()是对应位置相乘
@和torch.mm()是矩阵相乘 torch.bmm()相比mm增加了batch运算但不能广播
torch.matmul()相比bmm更是增加了广播,泛用性强