1. Element-wise Multiplication
*
torch.Tensor.mul()
torch.mul()
2. Matrix Multiplication
torch.Tensor.matmul()
torch.matmul()
torch.Tensor.mm()
torch.mm()
3. Batch Matrix Multiplication
torch.bmm()
torch.bmm(out_theta.transpose(1, 2), out_phi)
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:Multiplication in PyTorch - Python技术站