add comment explaining what this einsum does