I have a 1024x1024 image and I an AutoEncoder that gets patches of 128x128 as its input. I split the image into small patches (using numpy indexing) and get a list of all pa