torch Tensor.bmm()

Creator

Creator

Seonglae Cho

Created

Created

2024 Feb 24 15:55

Editor

Editor

Seonglae Cho

Edited

Edited

2024 May 25 4:47

Refs

Refs

Batch multiplication

@ is multiplication operator

If first dimension size is same, it is same with batch multiplication

This function do not

Pytorch Broadcasting

notion image

torch.bmm — PyTorch 2.3 documentation

Performs a batch matrix-matrix product of matrices stored in input and mat2.

torch.bmm — PyTorch 2.3 documentation

https://pytorch.org/docs/stable/generated/torch.bmm.html

Recommendations

/////////