如何在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, 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]))