Pytorch-有效的元素乘法？

Question

我有一个[100x3]的3D点张量

我有一个权重为[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

可以解决问题。