在把列表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)
2023-11-04 17:00:40
在把列表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)