Attention chunk axes inverted