ValueError: only one element tensors can be converted to Python scalars
1.先创建一个list,然后list append 的元素为tensor,再将list整体转为tensor时就会出现上述错,
解决办法:
1.先把list的tensor转换为numpy,然后再把list转换为tensor;
2.torch.tensor([item.cpu().detach().numpy() for item in 【list】]).to(self.device)
网友评论