PyTorch 教程


1 PyTorch 基础知识

    1.1 PyTorch 张量

    1.2 PyTorch Autograd 自动求导

    1.3 torch.nn 模块

torch.nn 模块

torch.nn

  • Module:创建一个行为类似于函数的可调用对象,但也可以包含状态(例如神经网络层权重)。 它知道其中包含的 Parameter ,并且可以将其所有坡度归零,遍历它们以进行权重更新等。
  • Parameter:张量的包装器,用于告知 Module 具有在反向传播期间需要更新的权重。 仅更新具有require_grad属性集的张量
  • functional:一个模块(通常按照惯例导入到 F 名称空间中),其中包含激活函数,损失函数等。 以及卷积和线性层等层的无状态版本。

参考:

https://pytorch.apachecn.org/#/docs/1.7/16