View source on GitHub
|
Outputs of a tfa.seq2seq.BasicDecoder step.
tfa.seq2seq.BasicDecoderOutput(
rnn_output, sample_id
)
Attributes | |
|---|---|
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.
|