使用RNN模型建模时,使用PackSequence处理不同长度的输入数据,作为RNN输入。
from torch.nn.utils.rnn import PackedSequence
对同一个patch中的不同文本使用padding的方式进行文本长度对齐,同时为了保证模型训练的精度,需要告诉RNN相关padding的情况,使用pack_padded_sequence实现。
import numpy as np
import torch
from torch.nn.utils.rnn import pack_padded_sequence, pad_packed_sequence
print(torch.__version__)
V = 5 # size of vocabulary
words = range(5)
data = [[1,2,3], [0,