没有softmax的tensorflow序列到序列

Question

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

谢谢

Answer 1

seq2seq.py 中的 model_with_buckets() 函数返回 2 个张量：输出和损失。 output 变量包含您正在寻找的解码器的原始输出（通常会被馈送到 softmax）。