... 共batch_size*num_beams個 beam_scores = torch.zeros((batch_size, ... 還沒結束,直接堆疊即可 decoded = torch.stack(best).type(torch.long).
確定! 回上一頁