我_s70041_ s softmax输出到预训练的分类器 . 然而，该分类器使用Keras嵌入层，因此将原始softmax传递到分类器中不是一种选择 . 我以为我可以使用gumbel softmax来获得一个热门编码然后使用我在这里找到的OneHotEmbedding层（https://github.com/keras-team/keras/issues/2505）来解决这个问题 .
Eric Jang为gumbel softmax提供了这个TensorFlow代码，我想知道如何将其转换为Keras层 . 特别是，我对 hard 属性感兴趣，该属性确保正向传递上的向量严格分类，但在向后传递时，渐变是gumbel softmax输出 . 我不知道如何在Keras Build 这个 . 有人可以帮忙吗？
谢谢 .