Calculate mutual information loss

same question as Mutual information as custom loss function