欢迎关注我的CSDN:https://blog.csdn.net/caroline_wendy
本文地址:https://blog.csdn.net/caroline_wendy/article/details/128222398
NLP seq2seq代码编写技巧
collate_fn
:pad_sequence
、 label=target[:,1:]
、 decoder_input=target[:, :-1]
torch.nn.utils.rnn.pad_sequence
collate_fn
输入是多个训练对,batch_size
个