python问个问题
我就写了121,130,131行，我的想法是先建立一个空列表，然后进入循环会得到一个张量，（heat_pred-heat）是一个size为（2,1,200,200）的张量，我想把张量里的结果都放到空列表里，但报错了，请问怎么解决
不是列表报错，是你取item就报错了
报错信息显示只有一个元素张量可以转变为标量
你参考129行和128行的代码，之前都是先取了个sum，你130行不取sum直接取item，应该是数量不对
我就写了121,130,131行，我的想法是先建立一个空列表，然后进入循环会得到一个张量，（heat_pred-heat）是一个size为（2,1,200,200）的张量，我想把张量里的结果都放到空列表里，但报错了，请问怎么解决
不是列表报错，是你取item就报错了
报错信息显示只有一个元素张量可以转变为标量
你参考129行和128行的代码，之前都是先取了个sum，你130行不取sum直接取item，应该是数量不对