关于extract.py这个函数的使用

[Danee-wawawa]

您好，非常感谢您的工作，在使用您提供的Visualization功能时，我在运行extract.py这个函数时，报了错误，其实就是被关掉的第18个问题。 提问者给我提供了一种解决方法，但我觉得会带来一些影响，而您说的“把DEEN所输出的三个分支的特征进行torch.cat()”的操作，是需要怎么改代码呢，我不太明白，希望能够得到您的帮助，感谢~

[ZYK100]

您好，您需要把3个分支的特征使用torch.cat()在通道维度连接为一个特征，比如：torch.cat((feat1,feat2,feat3),1)

[Danee-wawawa]

您是指将model.py文件里面的这个操作改为out = torch.cat((x, x1, x2), 1)吗？

[ZYK100]

是的

[Danee-wawawa]

您好，按照您的建议我对model.py进行了修改，但训练的时候还是out = torch.cat((x, x1, x2), 0)，只是训练结束，进行extract.py时，我改成了out = torch.cat((x, x1, x2), 1)，但是报了维度不匹配的错误，我想问下是训练的时候，就需要改成out = torch.cat((x, x1, x2), 1)吗？