Broadcasting

[davidweichiang]

torch.einsum now supports broadcasting: pytorch/pytorch#48420

[bdusell]

Yes, this would be a good feature to add, and I don't think it would be too tough to implement.

[davidweichiang]

Maybe just call torch.expand on the arguments beforehand?

[davidweichiang]

Or is it possible that it just works already, but the dimension checks are too strict?