nabu.neuralnetworks.components.beam_search_decoder.BeamSearchDecoderOutput Class Reference

class for the output of the BeamSearchDecoder More...

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

Detailed Description

class for the output of the BeamSearchDecoder

• predicted_ids: a [batch_size x beam_width] tensor containing the predicted ids
• parent_ids: a [batch_size x beam_width] tensor containing the id of the parent beam element

---

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

• components/beam_search_decoder.py