码迷,mamicode.com
首页 > 其他好文 > 详细

Multiplication in PyTorch

时间:2019-04-01 16:56:37      阅读:150      评论:0      收藏:0      [点我收藏+]

标签:pos   highlight   pytorch   out   nbsp   multi   bsp   matrix   batch   

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

标签:pos   highlight   pytorch   out   nbsp   multi   bsp   matrix   batch   

原文地址:https://www.cnblogs.com/hizhaolei/p/10637337.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!