def EntropicConfusion(features): batch_size = features.size(0) return torch.mul(features, torch.log(features)).sum() * (1.0 / batch_size)