tfa.seq2seq.BasicDecoderOutput

Outputs of a tfa.seq2seq.BasicDecoder step.

rnn_output The output for this step. If the output_layer argument of tfa.seq2seq.BasicDecoder was set, it is the output of this layer, otherwise it is the output of the RNN cell.
sample_id The token IDs sampled for this step, as returned by the sampler instance passed to tfa.seq2seq.BasicDecoder.