tf.compat.v1.nn.nce_loss