Multiplication in PyTorch

Posted hizhaolei

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了Multiplication in PyTorch相关的知识，希望对你有一定的参考价值。

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的主要内容，如果未能解决你的问题，请参考以下文章