tfa.seq2seq.BeamSearchDecoderOutput

View source on GitHub

Class BeamSearchDecoderOutput

BeamSearchDecoderOutput(scores, predicted_ids, parent_ids)

Aliases:

__new__

__new__(
    _cls,
    scores,
    predicted_ids,
    parent_ids
)

Create new instance of BeamSearchDecoderOutput(scores, predicted_ids, parent_ids)

Properties

scores

predicted_ids

parent_ids