欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

AttributeError: ‘list’ object has no attribute “dim”

程序员文章站 2022-05-27 11:51:22
...

这错是因为居然自己蠢到在pytorch用list当tensor用

将get_image转成tensor即可

get_image=torch.from_numpy(numpy.array(get_image))

这里我的get_image是list,所以要转tensor

相关标签: 深度学习

上一篇: python随机数

下一篇: 便捷操作