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 — длина последовательности, а * — любое количество конечных измерений, включая 0.

Для неотсортированных последовательностей используйте 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)

Параметры:

• sequences (list[Tensor]) – Список последовательностей, отсортированных по убыванию длины.
• enforce_sorted (bool, необязательно) – если True, проверяет, что входные последовательности отсортированы по убыванию длины. Если False, это условие не проверяется. По умолчанию: True.

Возвращает:

объект PackedSequence

Тип возвращаемого значения:

PackedSequence