matrix multiplication in pytorch

• *: element-wise multiplication

a = torch.tensor([[1, 2], [3, 4]])
b = torch.tensor([[2, 4], [6, 8]])
a*b

tensor([[ 2,  8],
        [18, 32]])

• broadcasting도 일어난다.

>>> A.size()
(131072, 3)
>>> B.size()
(131072, 1) 
>>> C = A * B 
>>> C.size()
(131072, 3)