torch.nn.utils.rnn.pad_sequence
时间: 2023-05-02 11:03:36 浏览: 153
python torch.utils.data.DataLoader使用方法
5星 · 资源好评率100%
torch.nn.utils.rnn.pad_sequence是一个PyTorch中的函数,用于将一批(batch)序列数据填充到相同长度。这个函数特别适用于RNN等模型的训练。当输入是一个sequence列表时,可以使用该函数将列表转换成一个Tensor。该函数会依据输入sequences中最长的序列来自动进行填充(padding),使得每个序列的长度相等。
阅读全文