test_vae_lstm.py 文件源码

def vae_loss(x, x_decoded_mean):
    x_d = Flatten()(x)
    x_dec_d = Flatten()(x_decoded_mean)
    xent_loss = input_dim * objectives.mean_squared_error(x_d, x_dec_d) 
    kl_loss = - 0.5 * K.sum(1 + z_log_var - K.square(z_mean) - K.exp(z_log_var), axis=-1) 
    return  xent_loss + kl_loss