torch.einsum ============ .. currentmodule:: torch .. autofunction:: einsum