[备忘录]pytorch中的batch Hadamard product,batch element wise product
pytorch针对batch只有矩阵乘法torch.bmm(),没有针对batch的Hadamard product,即不存在所谓的torch.bmul()之类的函数
可以使用万能的torch.einsum()公式
a=torch.arange(24).view(2, 3, 4) b=torch.ones_like(a) * 2 bmul = torch.einsum(\'ijk,ijk->ijk\',[a,b]) print(bmul)
torch.einsum()真是神器,附两个帖子
- 上一篇 »JavaScript备忘录,1——内置类型
- 下一篇 »PHP备忘录