Pytorch - 高效的elementwise乘法？

Question

我的3D点的张量[100x3]

我的权重[100X1]的向量，这就需要将元件明智乘以在X，Y，Z坐标。

目前，我创建一个新的向量W，我堆栈[100x3]元素与重复成[100x3]张量，以前我做的元素方式乘法。

我需要做很多次，这是太缓慢和内存密集型。有没有更好的办法？

Answer 1

在PyTorch标准乘法（*）已经是按元素。此外，它broadcasts。所以

import torch
xyz = torch.randn(100, 3)
w = torch.randn(100, 1)
multiplied = xyz * w

只会做的伎俩。