Я сейчас реализую модель seq-seq, и для декодера, когда она является логическим выводом, я бы хотел установить вывод текущего шага на вход следующего шага. Моя входная форма - пакетный * временной шаг * 102 димма, а для 102 дим - это горячая кодировка. Функция активации выхода декодера - softmax, и я хотел бы сделать его горячим для следующего входа декодера. Какую функцию я должен использовать для этого? Я попробовал tf.max и использовал tf.zero_like для построения тензора 0, обернутого переменной, и установил этот максимальный индекс равным единице, но это не удалось. Есть ли лучшее решение для этого?