torch.nn.utils.rnn.pack_sequence¶
- torch.nn.utils.rnn.pack_sequence(sequences, enforce_sorted=True)[源代码]¶
打包可变长度张量列表。
依次调用以下函数:
pad_sequence
、pack_padded_sequence
。sequences
应为大小为L x *
的张量列表,其中 L 是序列的长度,* 是任何数量的尾随维度,包括零。对于未排序的序列,使用 enforce_sorted = False。如果
enforce_sorted
为True
,则序列应按长度递减的顺序排序。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)
- 参数
- 返回值
一个
PackedSequence
对象- 返回类型