张量流序列到没有 softmax 的序列

Question

我正在使用 Tensorflow 序列对示例代码进行排序。出于某种原因，我不想将 softmax 添加到输出中。相反，我想在没有 softmax 的情况下获得解码器的原始输出。我想知道是否有人知道如何根据序列到序列示例代码来做到这一点？或者我需要从头开始创建它或修改 seq2seq.py（在 /tensorflow/tensorflow/python/ops/seq2seq.py 下）？

谢谢

Answer 1

model_with_buckets() 函数在 seq2seq.py returns 2 个张量：输出和损失。 outputs 变量包含您要查找的解码器的原始输出（通常会馈送到 softmax）。