面向 Tensorflow 的高效采样 softmax 实现
机器学习
2020-04-14 v1
摘要
本文讨论了一种用于 Tensorflow 的采样 softmax 损失（sampled softmax loss）的高效实现。相较于默认实现，其加速来源于对前向与反向传播计算图的简化。
关键词
引用
@article{arxiv.2004.05244,
title = {Efficient Sampled Softmax for Tensorflow},
author = {Maciej Skorski},
journal= {arXiv preprint arXiv:2004.05244},
year = {2020}
}