随笔分类 - Pytorch
摘要:torch.nn 是专门为神经网络设计的模块化接口,nn构建于autgrad之上,可以用来定义和运行神经网络 nn.Module 是nn中重要的类,包含网络各层的定义,以及forward方法 对于自己定义的网络,需要注意以下几点: 1)需要继承nn.Module类,并实现forward方法,只要在n
阅读全文
摘要:PyTorch中数据读取的一个重要接口是torch.utils.data.DataLoader,该接口定义在dataloader.py脚本中,只要是用PyTorch来训练模型基本都会用到该接口, 该接口主要用来将自定义的数据读取接口的输出或者PyTorch已有的数据读取接口的输入按照batch si
阅读全文
摘要:刚学习pytorch,简单记录一下 输出结果:
阅读全文
摘要:(1)生成一个未初始化的tensor (2)随机初始化一个tensor (3)矩阵的大小,直接将其当做元组来操作 (4)tensor之间的操作,可以有多重形式的操作,比如相加操作,直接+,或者add,等等 (5)tensor和numpy之间的呼唤,也是非常方便,除了chartensor之外,所有的t
阅读全文