数据集变换
(1)三维数据变四维
参考:https://www.jb51.net/article/167923.htm
数据为tensor形式:
1、压缩张量:torch.squeeze()
把维度数是1的消除
x.squeeze().size()
x=(1,3,4,5)=>(3,4,5)
2、扩充张量torch.unsqueeze()
x.unsqueeze(0).size()
——没使用成功,有报错,直接用的下面1、的方法
数据为narray形式:
1、a=np.expand_dims(a,axis=0)
实现(3,4,5)=>(1,3,4,5)
(2)数组转换(numpy中的narray和tensor相互转换)
参考:https://blog.csdn.net/weixin_43728604/article/details/102679016?ops_request_misc=%25257B%252522request%25255Fid%252522%25253A%252522160855410916780271145065%252522%25252C%252522scm%252522%25253A%25252220140713.130102334.pc%25255Fall.%252522%25257D&request_id=160855410916780271145065&biz_id=0&utm_medium=di