Function loss_fn

VAE_AutoEncoder/utils.py:101–110

(inputs,outputs,loss_fn,z_mean,z_log_var,num_features = 784)

    return z

def loss_fn(inputs,outputs,loss_fn,z_mean,z_log_var,num_features = 784):
    reconstruction_loss = loss_fn(outputs,inputs)
    reconstruction_loss = reconstruction_loss * num_features

    #计算KL散度损失值
    kl_loss = 1 + z_log_var - torch.square(z_mean) - torch.exp(z_log_var)
    kl_loss = -0.5 * torch.sum(kl_loss,dim = -1)
    vae_loss = torch.mean(reconstruction_loss + kl_loss)

    return vae_loss