tf.nn.weighted_cross_entropy_with_logits