Argmax Only Supported for AutoencoderKL

Introduction: Autoencoders are, in that respect, indispensable tools in the current realm of machine learning, particularly where data reduction or feature extraction is needed. The EMAE of the most discussed variant of the autoencoder comes from the adaptation of the Kullback-Leibler (KL) divergence loss function to the traditional autoencoder settings and has been named AutoencoderKL. … Read more