AI, Machine Learning, Artificial Intelligence, Deep Learning, Data Science.
upsampling layers of the decoder. The shape of the 3D tensor depends on the chosen architecture of the encoder and the size of the latent space. For example, if the final convolutional layer outputs feature maps of size 3x3x128, and the latent space has dimension 10, the flattened tensor would have