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)
上一篇:【报告分享】22020程序员人群洞察-Mob研究院(附下载)


下一篇:【报告分享】2021年“韭零后”基民人群洞察-Mob研究院(附下载)