torch.nn.utils.rnn.unpack_sequence

torch.nn.utils.rnn.unpack_sequence(packed_sequences)

將 PackedSequence 解包為可變長度 Tensor 列表。

packed_sequences 應該是一個 PackedSequence 物件。

示例

>>> from torch.nn.utils.rnn import pack_sequence, unpack_sequence
>>> a = torch.tensor([1, 2, 3])
>>> b = torch.tensor([4, 5])
>>> c = torch.tensor([6])
>>> sequences = [a, b, c]
>>> print(sequences)
[tensor([1, 2, 3]), tensor([4, 5]), tensor([6])]
>>> packed_sequences = pack_sequence(sequences)
>>> print(packed_sequences)
PackedSequence(data=tensor([1, 4, 6, 2, 5, 3]), batch_sizes=tensor([3, 2, 1]), sorted_indices=None, unsorted_indices=None)
>>> unpacked_sequences = unpack_sequence(packed_sequences)
>>> print(unpacked_sequences)
[tensor([1, 2, 3]), tensor([4, 5]), tensor([6])]

引數

packed_sequences (PackedSequence) – 一個 PackedSequence 物件。

返回值

一個 Tensor 物件列表

返回型別

list[torch.Tensor]