torch.
bmm
(input, mat2, deterministic=False, out=None) → Tensor
Performs a batch matrix-matrix product of matrices stored in input
and mat2
.
input
and mat2
must be 3-D tensors each containing the same number of matrices.
If input
is a (b×n×m) tensor, mat2
is a (b×m×p) tensor, out
will be a (b×n×p) tensor.
outi=inputi@mat2i