transpose in einsum, see features.py