Empty DimShuffles would return a scalar instead of an array, which can then lead to errors in ops that expect an array.