The current autoencoder and VAE only supports 2-d data, while there is a need to handle 3-d images. PR welcomed!
