如何在pytorch中批处理对话框数据集?

如何在pytorch中批处理对话框数据集?,pytorch,Pytorch,我想做一个面向任务的对话聊天机器人,用来预订餐厅。因为每个对话都有不同的顺序。有些人有5轮10句的对话,而另一个可能有6轮12句的对话,我不知道如何批处理数据集 你能给我一些教程或github示例吗?关于Stackoverflow,有一些相关的问题。我喜欢提供的解释/答案。将使用tldr版本。提供从链接复制的以下示例: a = [torch.tensor([1,2,3]), torch.tensor([3,4])] b = torch.nn.utils.rnn.pad_sequence(a, b

我想做一个面向任务的对话聊天机器人,用来预订餐厅。因为每个对话都有不同的顺序。有些人有5轮10句的对话,而另一个可能有6轮12句的对话,我不知道如何批处理数据集


你能给我一些教程或github示例吗?

关于Stackoverflow,有一些相关的问题。我喜欢提供的解释/答案。将使用tldr版本。提供从链接复制的以下示例:

a = [torch.tensor([1,2,3]), torch.tensor([3,4])]
b = torch.nn.utils.rnn.pad_sequence(a, batch_first=True)
>>>>
tensor([[ 1,  2,  3],
    [ 3,  4,  0]])
torch.nn.utils.rnn.pack_padded_sequence(b, batch_first=True, lengths=[3,2])
>>>>PackedSequence(data=tensor([ 1,  3,  2,  4,  3]), batch_sizes=tensor([ 2,  2,  1]))