关于tensor类型数据却报错‘list‘ object has no attribute ‘to‘

先放图:
在这里插入图片描述

问题所在:

我的数据本事应该是tensor类型,可以to(device),但是却报错是list(列表)所以无法’to’。

于是在出错前的语句先打印images,查明原因

print(images)

发现是[tensor[…], …],那问题很简单,在报错的语句前使用:

images = torch.stack(images, dim=1)

将列表元素取出(即list类型转换成tensor类型,将[tensor[....], ...]的数据改为tensor[....], ...),大功告成!

你可能感兴趣的:(pytorch,报错)