报错“ValueError: only one element tensors can be converted to Python scalar

在pytorch中,当想把一个list转换为tensor时,使用torch.tensor,有的时候会报“ValueError: only one element tensors can be converted to Python scalars”的错误。原因是:要转换的list里面的元素包含多维的tensor。解决方法如下:

val= torch.tensor([item.cpu().detach().numpy() for item in val]).cuda()

可能不需要cpu()或者cuda(),看自己的情况,val是要转换的list

你可能感兴趣的:(python,深度学习)