ValueError:only one element tensors can be converted to Python scalars

2021/5/7 22:25:24

本文主要是介绍ValueError:only one element tensors can be converted to Python scalars,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!

在把列表list转化为Tensor时,报错:

ValueError:only one element tensors can be converted to Python scalars

我的列表中每个元素类型都为tensor, 改正方法:
append 时将元素类型就改成数组nd.array形式
最后再转成tensor形式,就不报错了

images.append(img_i.cpu().numpy())
images = torch.tensor(images)


这篇关于ValueError:only one element tensors can be converted to Python scalars的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!


扫一扫关注最新编程教程