Is there a pytorch implementation of tf.nn.sampled_softmax_loss?

1

Hi everyone,

I was wondering whether there is a loss function having the same functionality as tf.nn.sampled_softmax_loss in tensorflow.

link: https://www.tensorflow.org/api_docs/python/tf/nn/sampled_softmax_loss

Thank you so much.

1 Like