深度学习,pytorch 学习笔记

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/MrWilliamVs/article/details/88567915

pytorch:
torch:主模块,用来搭建神经网络的。
torchvision: 辅模块,有数据库,和一些训练好的神经网络可以直接用。(VGG, AlexNet, ResNet)

和numpy相互转换,数据类型,运算。

变量Variable, 三部分:1。data,(相当于Tensor), 2. grad:(当前变量的梯度缓存区) 3. creator: 这个变量的创造节点。

Variable在计算时,后台默默搭建着一个庞大的系统,叫计算图(computational graph), 将所有的计算步骤(节点)都连接起来,最后,进行反向误差传递时,一次性将所有variable里面的修改幅度(梯度),都计算出来。
Tensor:只是相当于一个数据结构。

nn.Linear(in, out), in:每个输入样本的大小,(n, in), out:每个输出样本的小,(n, out)

猜你喜欢

转载自blog.csdn.net/MrWilliamVs/article/details/88567915
今日推荐