PackedSequence

class torch.nn.utils.rnn.PackedSequence(data, batch_sizes=None, sorted_indices=None, unsorted_indices=None) [source]

Содержит данные и список batch_sizes упакованной последовательности.

Все модули RNN принимают упакованные последовательности в качестве входных данных.

Примечание

Экземпляры этого класса никогда не должны создаваться вручную. Они предназначены для создания функциями, такими как pack_padded_sequence().

Размеры батчей представляют количество элементов на каждом шаге последовательности в батче, а не изменяющиеся длины последовательностей, передаваемые в pack_padded_sequence(). Например, для данных abc и x PackedSequence будет содержать данные axbc с batch_sizes=[2,1,1].

Переменные:

• data (Тензор) – Тензор, содержащий упакованную последовательность
• batch_sizes (Тензор) – Тензор целых чисел, содержащий информацию о размере батча на каждом шаге последовательности
• sorted_indices (Тензор, необязательно) – Тензор целых чисел, хранящий информацию о том, как эта PackedSequence построена из последовательностей.
• unsorted_indices (Тензор, необязательно) – Тензор целых чисел, хранящий информацию о том, как восстановить исходные последовательности в правильном порядке.

Примечание

data может находиться на произвольном устройстве и иметь произвольный тип данных. sorted_indices и unsorted_indices должны быть torch.int64 тензорами на том же устройстве, что и data.

Однако, batch_sizes всегда должен быть тензором на процессоре torch.int64.

Это условие сохраняется в классе PackedSequence, и во всех функциях, создающих :class:PackedSequence в PyTorch (т.е. они только передают тензоры, соответствующие этому условию).

property batch_sizes

Псевдоним для поля номер 1

count(value, /)

Возвращает количество вхождений значения.

property data

Псевдоним для поля номер 0

index(value, start=0, stop=9223372036854775807, /)

Возвращает первый индекс значения.

Вызывает ValueError, если значение отсутствует.

property is_cuda

Возвращает True, если self.data хранится на видеокарте

is_pinned() [source]

Возвращает True, если self.data хранится в закрепленной памяти

property sorted_indices

Псевдоним для поля номер 2

to(*args, **kwargs) [source]

Выполняет преобразование типа данных и/или устройства для self.data.

Она имеет аналогичную сигнатуру с torch.Tensor.to(), за исключением того, что необязательные аргументы, такие как non_blocking и copy должны передаваться в качестве аргументов-ключей, а не аргументов, иначе они не будут применены к тензорам индексов.

Примечание

Если тензор self.data уже имеет правильный torch.dtype и torch.device, то self возвращается. В противном случае возвращается копия с желаемой конфигурацией.

property unsorted_indices

Псевдоним для поля номер 3