Often times I get confused whether to use torch.matmul
or go for torch.bmm
I seriously don’t understand why do bmm
even exist. In what cases should we use torch.matmul
and in what cases should I use torch.bmm
?
You can always use torch.matmul
. torch.bmm
is a special case of torch.matmul
where both the tensors are 3-dimensional and contains equal number of matrices.
1 Like