Последовательность тензорного потока для последовательности без softmax

Question

Я использовал последовательность Tensorflow для примера кода последовательности. по какой-то причине я не хочу добавлять softmax для вывода. вместо этого я хочу получить необработанный вывод декодера без softmax. Мне было интересно, если кто-нибудь знает, как это сделать на основе последовательности для примера кода примера? Или мне нужно создать его с нуля или изменить seq2seq.py (в файле /tensorflow/tensorflow/python/ops/seq2seq.py)?

Спасибо Вам

Answer 1

The model_with_buckets() функция в seq2seq.py возвращает 2 тензоров: выход и потери. Переменная выходов содержит исходный вывод декодера, который вы ищете (который обычно подается на softmax).