pytorch基础操作
作者:互联网
1、 view()的作用相当于numpy中的reshape,重新定义矩阵的形状
x=x.view(-1,32*5*5)#view的功能跟reahape相同,在这里的作用是展平,与torch.flatten功能相同
2、 降维torch.squeeze(input, dim=None, out=None)
3、 增维torch.unsqueeze(input, dim, out=None)
4、安装 pycocotools
Linux: pip install pycocotools
Windows:pip install pycocotools-windows
5、totensor的作用:将PIL图像或者一个numpy的数据转换为tensor格式,将[高度,宽度,通道]转换为[通道,高度,宽度],并将[0,255]之间的像素值映射到[0,1]
import torchvision.transforms as transforms
transform=transforms.Compose([
transforms.ToTensor(),
transforms.Normalize((0.5,0.5,0.5),(0.5,0.5,0.5))
])
6、5中Normalize的作用:使用均值和标准差对tensor进行标准化
标签:None,0.5,torch,基础,pytorch,transforms,pycocotools,操作,view 来源: https://blog.csdn.net/weixin_43745234/article/details/119911392