1、在张量的后面加了(),有的话,则删除,如果没有加的话,可以debug或者打印它的type,查看是否都是tensor。
2、查看loss = loss_fn(y_pre, y),变量名loss是否和函数名loss_fn相同,即可能都写成了loss,那么也会出现这种报错。

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐