PackedSequence

class torch.nn.utils.rnn.PackedSequence(data, batch_sizes=None, sorted_indices=None, unsorted_indices=None) [исходный код]

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

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

Примечание

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

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

Переменные:

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

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

Self

Примечание

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

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

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

batch_sizes: Tensor

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

count(value, /)

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

data: Tensor

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

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

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

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

property is_cuda: bool

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

is_pinned() [исходный код]

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

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

bool

sorted_indices: Tensor | None

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

to(*args, **kwargs) [исходный код]

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

Сигнатура этого метода аналогична сигнатуре torch.Tensor.to()

Примечание

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

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

Self

unsorted_indices: Tensor | None

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