decoder_beam_search_output: Beam Search Decoder Output

In tfaddons: Interface to 'TensorFlow SIG Addons'

Description

Beam Search Decoder Output

Usage

decoder_beam_search_output(scores, predicted_ids, parent_ids)

Arguments

scores	calculate the scores for each beam
predicted_ids	The final prediction. A tensor of shape '[batch_size, T, beam_width]' (or '[T, batch_size, beam_width]' if 'output_time_major' is 'TRUE'). Beams are ordered from best to worst.
parent_ids	The parent ids of shape '[max_time, batch_size, beam_width]'.

Value

None

tfaddons documentation built on July 2, 2020, 2:12 a.m.