数据预处理

list 和 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()