array_zip works for multiple arrays; broadcastShapes