Pytorch Broadcasting torch.matmul — PyTorch 2.3 documentationThe behavior depends on the dimensionality of the tensors as follows:https://pytorch.org/docs/stable/generated/torch.matmul.html#torch.matmul