(self, beam_size)
Repeat beam_size times along batch dimension.
Repeat beam_size times along batch dimension.
Repeat beam_size times along batch dimension .
def repeat_beam_size_times(self, beam_size):
        """ Repeat beam_size times along batch dimension. """
        for model_state in self.model_decoder_states:
            model_state.repeat_beam_size_times(beam_size)
def repeat_beam_size_times ( self , beam_size ) : for model_state in self . model_decoder_states : model_state . repeat_beam_size_times ( beam_size )