Followup to Scatter Layout normalization
Followup to Scatter Layout normalization

The code already implicitly assumes that ScatterSimplifier has run before. We
cannot normalize if there are more than 1 "scatter" (batch) dimensions. Add a
check for that and adjust a test case that would in fact be incorrectly
normalized (but hidden from the verifier by running ScatterSimplifier
afterwards).
