I would like to train an autoencoder (with convolutional blocks) on a massive amount of heat maps using Keras. The heat maps are 2D and of size 1000 x 1000 x 1 (one color channe