pytorch学习第一天

torchvision

为了方便加载以上五种数据库的数据,pytorch团队帮我们写了一个torchvision包。使用torchvision就可以轻松实现数据的加载预处理
我们以使用CIFAR10为例:
导入torchvision的库

import torchvision
import torchvision.transforms as transforms  # transforms用于数据预处理
def imshow(img):
    img = img / 2 + 0.5 #unnormalize
    npimg = img.numpy()
    plt.imshow(np.transpose(npimg, (1, 2, 0)))

imshow(torchvision.utils.make_grid(images))

torch.utils.data

dataset (Dataset) – 加载数据的数据集。
batch_size (int, optional) – 每个batch加载多少个样本(默认: 1)。
shuffle (bool, optional) – 设置为True时会在每个epoch重新打乱数据(默认: False).
sampler (Sampler, optional) – 定义从数据集中提取样本的策略。如果指定,则忽略shuffle参数。
num_workers (int, optional) – 用多少个子进程加载数据。0表示数据将在主进程中加载(默认: 0)
drop_last (bool, optional) – 如果数据集大小不能被batch size整除,则设置为True后可删除最后一个不完整的batch。如果设为False并且数据集的大小不能被batch size整除,则最后一个batch将更小。(默认: False)
数据加载器,选择

torch.nn 与 torch.nn.functional的区别

Torch

torch会将torch产生的tensor放在GPU中加速运算,就像numpy会把array放在cpu中加速运算一样。另外torch做的和numpy能很好兼容,这样就可以自由的转换numpy array和torch tensor了。
可将torch视作numpy去思考:

np.mean(data)
torch.mean(tensor)

torch张量

torch.is_tensor()
a = torch.randn(1,2,3,4,5)
torch.numel(a)
>>>120

将numpy.narray转成Tensor()

a = np.array([1,2,3])
t = torch.from_numpy(a)

技巧:视torch为np,具体有哪些方法,随用随查。

©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

  • 每过一段时间,总会有一个python库被开发出来,改变深度学习领域。而PyTorch就是这样一个库。 在过去的几周...
    AiTechYun阅读 4,278评论 0 4
  • torch 本笔记引用自PyTorch中文文档 包torch包含了多维疑是的数据结构及基于其上的多种数学操作。 1...
    zhaoQiang012阅读 84,037评论 0 27
  • Spring Cloud为开发人员提供了快速构建分布式系统中一些常见模式的工具(例如配置管理,服务发现,断路器,智...
    卡卡罗2017阅读 136,665评论 19 139
  • 现在,很多孕妇都格外重视怀孕期间的营养补充,动辄鲍参翅肚齐上阵,稍不留神就会造成营养过盛。作为一类特殊人群,孕妇在...
    易明达海参阅读 563评论 0 0
  • 耶路撒冷是信徒向往的朝拜圣地。曾经却是内些阿拉伯人和犹太人心中的噩梦。殖民主义嘲讽着这个神圣之地,杀戮,血液,红...
    北斋俉生阅读 306评论 0 0

友情链接更多精彩内容