torch.squeeze() 这个函数主要对数据的维度进行压缩,去掉维数为1的的维度,设图像大小为(1,36,60),可以去掉,变为(36,60)
torch.fromnumpy()
torch.unsqueeze()都数据进行扩充(36,60),扩充为(1,36,60)
torch.squeeze() 这个函数主要对数据的维度进行压缩,去掉维数为1的的维度,设图像大小为(1,36,60),可以去掉,变为(36,60)
torch.fromnumpy()
torch.unsqueeze()都数据进行扩充(36,60),扩充为(1,36,60)