关于#ar#的问题：PyTorch进行深度学习任务解决VRP问题(语言-python)

Question

下面是原代码。

PyTorch进行深度学习任务解决VRP问题。

Answer 1

引自免费微信小程序：皆我百晓生

看起来你在尝试使用PyTorch实现一个基于Transformer的Vehicle Routing Problem（VRP）模型，并遇到了一个类型错误。根据报错信息，问题出在view()函数调用上，它期望接收一个元组作为参数，但接收到的是列表。具体来说，在这行代码：

x = x.view(batch, city_size, city_size, M, -1)

这里的M应该是一个整数，而不是一个列表。为了解决这个问题，你需要确保M是一个整数。通常在定义模型时，你会明确设置这个值，比如M = 8。如果你是从某个变量获取M，确保在使用之前它是整型。

这里是修复后的代码段：

# ... 其他代码 ...

# 确保M是整数
M = 8

# 第一层MHA
queryl = self.wql(node)
queryl = t.unsqueeze(queryl, dim=2)
query1 = queryl.expand(batch, city_size, city_size, M, embedding_size)

# ... 其他代码 ...

如果你的M确实来自一个可能不是整数的地方，那么在使用前添加类型检查并转换为整数：

if isinstance(M, list) or isinstance(M, tuple):
    M = int(M[0])

然后将修复后的M用于上述view()函数。如果还有其他错误，或者你需要更详细的解释，请告诉我。这段代码只是一个指导，实际运行之前，你需要确保整个上下文环境正确无误。