Beam search

[antonio-mastropaolo]

Hello everyone,

I was wondering if we could add an option when getting the prediction such that instead of having only the most likely one among the explored beams, it returns all the explored paths.

Thanks for any further feedback