Adds comment about shapes in beam_search_helper.py.
Adds comment about shapes in beam_search_helper.py.

PiperOrigin-RevId: 216807996

tensorflow/lingvo,tensorflow/lingvo,tensorflow/lingvo,tensorflow/lingvo