该方法会返回一个PackedSequence对象,其中包含data保存拉平的数据 和batch_sizes保存时间步相应的批次大小,比如上面就是tensor([4, 3, 3, 2, 1, 1])。 Pytorch的RNN(LSTM/GRU)可以接收PackedSequence,并返回一个新的PackedSequence。然后我们可以用pad_packed_sequence方法把返回的PackedSequence还原成我们想要的形式。
首先,packed是包装好的的意思;padded是填充的意思;pack有包装、压紧的意思;pad有填充的意思。 pack_padded_sequence即压紧封装填充好的句子 pad_packed_sequence即填充被压紧封装好的句子 示意图: pack_padded_sequence pad_packed_sequence就是上边的那个过程反过来 pack_padded_sequence #官方函数torch.nn.utils.rnn...
hidden_size, num_layers)# 将 packed_seqs 传递给 LSTM 模型packed_output, ht, ct=model(packed_seqs)# 使用 pad_packed_sequence 将输出转换回填充的序列格式unpacked_output
(sequence_length, batch_size, embedding_size),就是变成一列是一个句子,据说这样可以提高并行处理效率,但是对于我这种新手来说形状有点抽象,所以老老实实用False了。 之后进行pack操作,pack之后变成PackedSequence类型,就能实现上文说的padding位对模型无效的效果了 batch = torch.nn.utils.rnn.pack_padded_sequence...
在使用 pytorch 的 RNN 模块的时候, 有时会不可避免的使用到pack_padded_sequence和pad_packed_sequence,当使用双向RNN的时候, 必须要使用 pack_padded_sequence !!.否则的话, pytorch 是无法获得 序列的长度, 这样也无法正确的计算双向RNN/GRU/LSTM的结果. ...
更重要的是,这种压制成(packed)的输入数据,pytorch中的RNN,LSTM模型都是认的,只需要把它们直接作为Module的input输入进去,可以得到对应的PackedSequence。 最后,我们可以使用pad_packed_sequence来反向把PackSequence变回padded后的矩阵和序列长度。 >>>fromtorch.nn.utils.rnnimportpad_packed_sequence>>>packed_seq=pa...
这里补充下对Pytorch中pack_padded_sequence和pad_packed_sequence的理解。 当我们训练RNN时,如果想要进行批次化训练,就得需要截断和填充。 因为句子的长短不一,一般选择一个合适的长度来进行截断; 而填充是在句子过短时,需要以 填充字符 填充,使得该批次内所有的句子长度相同。
但是pack_sequence这玩意返回的PackedSequence是batch_first=False的,这就让我很不爽,这意味着我的RNN必须是batch_first=False,所以不得不重新定义网络 rnn = nn.RNN(1, 3, batch_first=False) print(rnn(packed_sequence)) # 搞定! 最后,我...
torch.nn.utils.rnn.pack_padded_sequencetorch.nn.utils.rnn.pad_packed_sequence计算图: 1) 把词的ID通过Embedding层变成向量。 2) 把padding后的数据进行pack。 3) 传入GRU进行Forward计算。 4) Unpack计算结果 5) 把双向GRU的结果向量加起来。 6) 返回(所有时刻的)输出和最后时刻的隐状态。 输入: input_...
# 而如果输入是PackedSequence对象,那么输出outputs也是一个PackedSequence对象,我们需要用 # 函数pad_packed_sequence把它变成一个shape为(max_length, batch, hidden*num_directions)的向量以及 # 一个list,表示输出的长度,当然这个list和输入的input_lengths完全一样,因此通常我们不需要它。