How to use the value 'decoded_batch' in function 'def beam_decode' when I test my own model? 