torch.nn.utils.rnn.pack_padded_sequence

torch.nn.utils.rnn.pack_padded_sequence(input, lengths, batch_first=False, enforce_sorted=True)

打包包含可变长度填充序列的张量。

input 的大小可以为 T x B x *（如果 batch_first 为 False）或 B x T x *（如果 batch_first 为 True），其中 T 是最长序列的长度，B 是批大小，* 是任意数量的维度（包括 0）。

对于未排序的序列，使用 enforce_sorted = False。如果 enforce_sorted 为 True，则序列应按长度降序排序，即 input[:,0] 应是最长序列，input[:,B-1] 是最短序列。enforce_sorted = True 仅在 ONNX 导出时才需要。

注意

此函数接受至少具有两个维度的任何输入。您可以应用它来打包标签，并使用 RNN 的输出与它们一起直接计算损失。可以通过访问其 .data 属性从 PackedSequence 对象中检索张量。

参数

• input (张量) – 可变长度序列的填充批。

• lengths (张量 或 列表(整数)) – 每个批元素的序列长度列表（如果作为张量提供，则必须在 CPU 上）。

• batch_first (布尔值, 可选) – 如果 True，则期望输入采用 B x T x * 格式，否则为 T x B x *。

• enforce_sorted (布尔值, 可选) – 如果 True，则期望输入包含按长度降序排序的序列。如果 False，则输入将无条件排序。默认值：True。

返回值

一个 PackedSequence 对象

返回类型

PackedSequence