Efficient Sampled Softmax for Tensorflow

04/10/2020

∙

by   Maciej Skorski, et al.

∙

0

∙

share

This short paper discusses an efficient implementation of sampled softmax loss for Tensorflow. The speedup over the default implementation is achieved due to simplification of the graph for the forward and backward passes.