PyTorch element-wise product of vectors / matrices / tensors

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().