1. 点乘
element-wise multiplication
可以用torch.mul(a, b)实现;
也可以直接用*实现。
2. 矩阵乘
torch.mm只针对二维矩阵
torch.matmul是tensor的矩阵乘法。当输入是都是二维时,就是矩阵乘法。
torch.bmm是tensor的矩阵乘法,两个tensor的维度必须为3。
1. 点乘
element-wise multiplication
可以用torch.mul(a, b)实现;
也可以直接用*实现。
2. 矩阵乘
torch.mm只针对二维矩阵
torch.matmul是tensor的矩阵乘法。当输入是都是二维时,就是矩阵乘法。
torch.bmm是tensor的矩阵乘法,两个tensor的维度必须为3。