在把列表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)