tf.compat.v1.nn.sampled_softmax_loss