编程语言
首页 > 编程语言> > ValueError:only one element tensors can be converted to Python scalars

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)

标签:tensor,Python,element,scalars,报错,images,ValueError,append
来源: https://blog.csdn.net/qq_35037684/article/details/116503607