随笔分类 -  Pytorch

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

点击右上角即可分享
微信分享提示