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进行标准化

上一篇:No module named 'pycocotools': 一行解决 Windowns下安装pycocotools 简单完美


下一篇:No module named ‘pycocotools‘