The loss function first computes binary cross entropy loss between the source x and the reconstructed x and stores that single tensor value as bce. ... Listing 3: Training a Variational Autoencoder.