from_dist_to_loss_torch function

Function to transform a distribution layer output into a loss function