Note this is only available in the pytorch variant [1], not the original numpy interface that it is based on [2] [1] https://pytorch.org/docs/stable/generated/

Note this is only available in the pytorch variant [1], not the original numpy interface that it is based on [2] [1] https://pytorch.org/docs/stable/generated/torch.einsum.html https://pytorch.org/docs/stable/generated/torch.einsum.html [2] https://numpy.org/doc/stable/reference/generated/numpy.einsum.html https://numpy.org/doc/stable/reference/generated/numpy.einsu...