torch
ShuqiaoS
这个作者很懒,什么都没留下…
展开
-
【文档学习】PyTorch——torch包
本系列记录了博主学习PyTorch过程中的笔记。本文记录的是torch包的相关内容,官方网址。只记录了一部分博主用到的,持续更新。更新于2019.03.21。torch包包括了用于多维张量的数据结构,并定义了对于它们的数学计算。此外,这个包也提供了多种有效的张量及任意类型的并行化用法,以及其他有用的方法。这个包对应于CUDA,允许用户在NVIDIA GPU(运算能力>=3.0)上运行张量...原创 2019-03-21 11:03:56 · 8621 阅读 · 0 评论 -
【PyTorch入门】(三)神经网络(Neural Networks)
官网。更新于2019.06.13。文章目录网络定义损失函数BackpropUpdate the weights可以用torch.nn包构建神经网络。看完这篇文章应该对autograd有了一定的了解,nn是基于autograd定义和求导模型的。一个nn.Module包括层以及用于返回output的forward(input)方法。比如,对于如下数字图像分类网络(convnet):这是一个简...原创 2019-06-13 16:57:13 · 1192 阅读 · 0 评论 -
【文档学习】Pytorch——torch.autorgrad包
torch.autograd提供了实现自动计算任意标量值函数的类别核函数,需要手动修改现有代码(需要重新定义需要计算梯度Tensor,加上关键词requires_grad=True)。本文更新于2019.06.12。文章目录torch.autograd.backwardtorch.autograd.grad局部梯度计算失效(Locally disabling gradient computati...原创 2019-06-12 13:45:51 · 4163 阅读 · 0 评论