class for the final output of the BeamSearchDecoder More...

Inheritance diagram for nabu.neuralnetworks.components.beam_search_decoder.BeamSearchDecoderFinalOutput:

Detailed Description

class for the final output of the BeamSearchDecoder

predicted_ids: a [time x batch_size x beam_width] tensor containing the predicted ids
lengths: a [1 x batch_size x beam_width] tensor containing the lengths
scores: a [1 x batch_size x beam_width] tensor containing the scores
alignments: a [time x batch_size x beam_width x in_time] tensor containing the alignments

The documentation for this class was generated from the following file: