Seonglae ChoBatch multiplication
@ is multiplication operator
If first dimension size is same, it is same with batch multiplication
This function do not Pytorch Broadcasting
torch.bmm — PyTorch 2.3 documentation
Performs a batch matrix-matrix product of matrices stored in input
and mat2.
https://pytorch.org/docs/stable/generated/torch.bmm.html