快捷方式

torch.nn.utils.rnn.pack_sequence

torch.nn.utils.rnn.pack_sequence(sequences, enforce_sorted=True)[源代码]

打包可变长度张量列表。

依次调用以下函数:pad_sequencepack_padded_sequence

sequences 应为大小为 L x * 的张量列表,其中 L 是序列的长度,* 是任何数量的尾随维度,包括零。

对于未排序的序列,使用 enforce_sorted = False。如果 enforce_sortedTrue,则序列应按长度递减的顺序排序。 enforce_sorted = True 仅在 ONNX 导出时才必要。

示例

>>> from torch.nn.utils.rnn import pack_sequence
>>> a = torch.tensor([1, 2, 3])
>>> b = torch.tensor([4, 5])
>>> c = torch.tensor([6])
>>> pack_sequence([a, b, c])
PackedSequence(data=tensor([1, 4, 6, 2, 5, 3]), batch_sizes=tensor([3, 2, 1]), sorted_indices=None, unsorted_indices=None)
参数
  • sequences (list[Tensor]) – 按长度递减的顺序排列的序列列表。

  • enforce_sorted (bool, 可选) – 如果 True,则检查输入是否包含按长度递减的顺序排序的序列。如果 False,则不会检查此条件。默认值:True

返回值

一个 PackedSequence 对象

返回类型

PackedSequence

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源