1.问题说明 在处理数据的过程中,一个batch中数据邻接矩阵维度不同,例如,batch=2(表示一个批次两个句子),对应邻接矩阵维度分别为22,44.此时无法在一个batch中进行处理。 2.解决方案 选择最大长度句子为矩阵形状,对小于该长度的矩阵进行填充。使用torch.nn.torch.nn.ZeroPad2d(padding=(0, 2, 0, 2)) 如下: a = torch.ones(2, 2) pad = torch.nn.ZeroPad2d(padding=(0