torch - 如何做Elementwise产品 torch ？

Question

如何找到两个张量的元素乘积？我的两个张量具有相同的维度，我想找到它们的乘积？

q
 1  2  3
 2  4  6

w
 1  2  3
 2  4  6

它应该产生:

1   4   9
4  16  36

Answer 1

您可以使用 cmul 来完成此操作。

th> torch.cmul(q,w)
  1   4   9
  4  16  36

附注:

q:cmul(w):将它们相乘并将值存储回 q，

z=torch.cmul(q,w):将它们相乘并返回一个新的张量，该张量将存储在 z 中。