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

时间:2016-04-21 03:24:27

标签: python tensorflow

我使用Tensorflow序列来排序示例代码。出于某种原因,我不想将softmax添加到输出中。相反,我想获得没有softmax的解码器的原始输出。我想知道是否有人知道如何根据序列来排序示例代码?或者我需要从头开始创建它或修改seq2seq.py(在/tensorflow/tensorflow/python/ops/seq2seq.py下)?

谢谢

1 个答案:

答案 0 :(得分:1)

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