This circumvents a bug when DimShuffle of a scalar shows up inside a Blockwise, as the outer indexing yields a float (as opposed to a numpy scalar) which has no `.shape` attribute.