This isn't correct. It's the shape of the stack, but it's not the shape of the resulting ArrayStack. For example, stacking 6 Arrays, each of which has shape (300, 400), in a (3, 2) pattern will result in an ArrayStack with shape (3, 2, 300, 400).