tf.compat.v1.nn.weighted_cross_entropy_with_logits