一.数据类型转换
1.pytorch的tensor类型转numpy类型
numpy数据类型=tensor.numpy(tensor数据类型)
2.numpy类型转pytorch的tensor类型
tensor数据类型=torch.from_numpy(numpy数据类型)
3.python的list类型转pytorch的tensor类型
暂时还没有遇到
4.pytorch的tensor类型转python的list类型
tensor数据类型=torch.Tensor(list数据类型)
5.tensor数据类型的device由cpu转cuda
tensor(cpu中数据)=tensor数据类型.cpu()
6.tensor数据类型的device由cuda转cpu
tensor(cuda中数据)=tensor数据类型.cuda()
二.数据维度
1.图像image,[1, 3, 500, 355] ---> [3, 500, 355],对数据维度进行压缩
image = image[0]
2.图像image,[3, 500, 355] ---> [1, 3, 500, 355],对数据维度进行扩充
image = image.unsqueenze_()