我使用Tensorflow序列来排序示例代码。出于某种原因,我不想将softmax添加到输出中。相反,我想获得没有softmax的解码器的原始输出。我想知道是否有人知道如何根据序列来排序示例代码?或者我需要从头开始创建它或修改seq2seq.py(在/tensorflow/tensorflow/python/ops/seq2seq.py下)?
谢谢
答案 0 :(得分:1)
seq2seq.py中的model_with_buckets()函数返回2个张量:输出和损失。输出变量包含您正在寻找的解码器的原始输出(通常会馈送到softmax)。