torch.nn.utils.rnn.unpack_sequence#

torch.nn.utils.rnn.unpack_sequence(packed_sequences)[source]#

Unpack PackedSequence into a list of variable length Tensors.

packed_sequences should be a PackedSequence object.

Example

>>>fromtorch.nn.utils.rnnimportpack_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])]

Parameters

packed_sequences (PackedSequence) – A PackedSequence object.

Returns

a list ofTensor objects

Return type

list[torch.Tensor]