transformers

aligned sample_beam output selection with beam_search

#25375

Merged