![](https://img-blog.csdnimg.cn/20201014180756780.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch学习
文章平均质量分 91
MapleTx
这个作者很懒,什么都没留下…
展开
-
pytorch官方教程笔记--01基本概念与自动微分
基础1 入门 pytorch基础概念,类似于numpy的ndarrays,好处是可以使用GPU进行计算 构造矩阵的不同方式: from __future__ import print_function import torch x=torch.empty(5,3) print(x) x=torch.rand(5,3) print(x) x=torch.zeros(5,3,dtype=torch.long) print(x) x=torch.tensor([5.56,5,3],dtype=float) pr原创 2020-09-15 17:37:03 · 129 阅读 · 0 评论 -
pytorch官方教程笔记--02神经网络
神经网络 神经网络可以通过 torch.nn 包来构建。一个 nn.Module 包括层和一个方法 forward(input) 它会返回输出(output)。 例如,看一下数字图片识别的网络: 典型的神经网络训练过程包括以下几点: 1.定义一个包含可训练参数的神经网络 2.迭代整个输入 3.通过神经网络处理输入 4.计算损失(loss) 5.反向传播梯度到神经网络的参数 6.更新网络的参数,典型的用一个简单的更新方法:weight = weight - learning_rate *gradient原创 2020-10-05 11:07:43 · 294 阅读 · 0 评论