Pytorch - Эффективное поэлементное умножение?

Question

У меня есть тензор трехмерных точек [100x3]

У меня есть вектор весов [100x1], который нужно умножить на элемент в координаты X, Y, Z.

В настоящее время я создаю новый вектор W, в котором я складываю элемент [100x3] с повторением в тензор [100x3], прежде чем выполнить умножение на элемент.

Мне нужно сделать это много рази это слишком медленно и требует много памяти.Есть ли лучший способ?

Answer 1

Стандартное умножение (*) в PyTorch уже поэлементно.Дополнительно транслирует .Так что

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

просто сделает свое дело.