expand teacher prompt tensors by num_generations in _prepare_teacher_logprob_inputs#122

zhuxiaoxuhit wants to merge 1 commit intowenet-e2e:mainwenet-e2e/west:mainfrom

zhuxiaoxuhit:fix/teacher-logprob-inputs-num-generationszhuxiaoxuhit/west:fix/teacher-logprob-inputs-num-generationsCopy head branch name to clipboard