I would extend this inline comment to tell that using `einsum` is useful  at sparing a temporary memory allocation.