其他分享
首页 > 其他分享> > pytorch基础操作

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