tfa.seq2seq.BeamSearchDecoderState

View source on GitHub

BeamSearchDecoderState(cell_state, log_probs, finished, lengths, accumulated_attention_probs)

cell_state

log_probs

finished

lengths

accumulated_attention_probs