快捷方式

torch.nn.utils.rnn.unpad_sequence

torch.nn.utils.rnn.unpad_sequence(padded_sequences, lengths, batch_first=False)[源代码]

将填充的张量解填充为可变长度张量的列表。

unpad_sequence 将填充的张量解填充为可变长度张量的列表。

示例

>>> from torch.nn.utils.rnn import pad_sequence, unpad_sequence
>>> a = torch.ones(25, 300)
>>> b = torch.ones(22, 300)
>>> c = torch.ones(15, 300)
>>> sequences = [a, b, c]
>>> padded_sequences = pad_sequence(sequences)
>>> lengths = torch.as_tensor([v.size(0) for v in sequences])
>>> unpadded_sequences = unpad_sequence(padded_sequences, lengths)
>>> torch.allclose(sequences[0], unpadded_sequences[0])
True
>>> torch.allclose(sequences[1], unpadded_sequences[1])
True
>>> torch.allclose(sequences[2], unpadded_sequences[2])
True
参数
  • padded_sequences (张量) – 填充的序列。

  • lengths (张量) – 原始(未填充)序列的长度。

  • batch_first (布尔值, 可选) – 批次维度是否为第一维度。默认值:False。

返回值

Tensor 对象列表

返回类型

列表[张量]

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源