arXivDaily arXiv每日学术速递 周一至周五更新
北美退伍军人节(Veterans Day),12 日无更新
arXiv 2004.05244cs.LG
收藏

用于Tensorflow的高效采样Softmax

Efficient Sampled Softmax for Tensorflow

Maciej Skorski

摘要:

这篇简短的论文讨论了Tensorflow的\emph{sampled softmax loss}的高效实现。由于简化了向前和向后传递的图形,因此实现了对默认实现的加速。

英文摘要:

This short paper discusses an efficient implementation of \emph{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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑