Understanding NumPy's einsum Understanding NumPy's einsum python python