学习笔记｜PackedSequence

使用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,