diff options
| author |  Adam Roberts <adarob@google.com> | 2018-01-04 11:04:10 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-01-04 11:11:31 -0800 |
| commit | 2f36b8a0e5c61c24e3754c8227494275a7e9dbc4 (patch) | |
| tree | 743cca32b416a66027940f7feaf5bd851cba41b2 /tensorflow/contrib/seq2seq | |
| parent | b6078bc350acd304a2db09ad4deb2d09fab2178b (diff) | |
Clarify confusing documentation regarding FinalBeamSearchDecoderOutput.
PiperOrigin-RevId: 180820115
Diffstat (limited to 'tensorflow/contrib/seq2seq')
| -rw-r--r-- | tensorflow/contrib/seq2seq/python/ops/beam_search_decoder.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/contrib/seq2seq/python/ops/beam_search_decoder.py b/tensorflow/contrib/seq2seq/python/ops/beam_search_decoder.py index 5be0c92243..1ff3df4c22 100644 --- a/tensorflow/contrib/seq2seq/python/ops/beam_search_decoder.py +++ b/tensorflow/contrib/seq2seq/python/ops/beam_search_decoder.py @@ -67,7 +67,8 @@ class FinalBeamSearchDecoderOutput( Args: predicted_ids: The final prediction. A tensor of shape - `[T, batch_size, beam_width]`. + `[batch_size, T, beam_width]` (or `[T, batch_size, beam_width]` if + `output_time_major` is True). Beams are ordered from best to worst. beam_search_decoder_output: An instance of `BeamSearchDecoderOutput` that describes the state of the beam search. """ |