torch.nn.utils.rnn.pad_sequence

torch.nn.utils.rnn.pad_sequence(sequences, batch_first=False, padding_value=0.0, padding_side='right') [source]

Дополняет список тензоров переменной длины значением padding_value.

pad_sequence объединяет список тензоров вдоль нового измерения и дополняет их до одинаковой длины. sequences может быть списком последовательностей размера L x *, где L — длина последовательности, а * — любое число измерений (включая 0). Если batch_first равно False, выход имеет размер T x B x *, а в противном случае — B x T x *, где B — размер пакета (число элементов в sequences), а T — длина самой длинной последовательности.

Пример

>>> from torch.nn.utils.rnn import pad_sequence
>>> a = torch.ones(25, 300)
>>> b = torch.ones(22, 300)
>>> c = torch.ones(15, 300)
>>> pad_sequence([a, b, c]).size()
torch.Size([25, 3, 300])

Примечание

Эта функция возвращает Tensor размером T x B x * или B x T x *, где T — длина самой длинной последовательности. Предполагается, что конечные измерения и тип всех тензоров в последовательностях совпадают.

Параметры:

• sequences (list[Tensor]) – список последовательностей переменной длины.
• batch_first (bool, optional) – если True, выход будет в формате B x T x *, в противном случае — T x B x *.
• padding_value (float, optional) – значение для элементов заполнения. По умолчанию: 0.
• padding_side (str, optional) – сторона, с которой дополняются последовательности. По умолчанию: 'right'.

Возвращает:

Tensor размером T x B x *, если batch_first равно False. В противном случае Tensor размером B x T x *

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

Tensor