In PyTorch, how do I get the element-wise product of two vectors / matrices / tensors?
For googlers, this is product is also known as:
Given two tensors A
and B
you can use either:
A * B
torch.mul(A, B)
A.mul(B)
Note: for matrix multiplication, you want to use A @ B
which is equivalent to torch.matmul()
.