torch.nn 介绍(部分)

翻译自:https://pytorch.org/docs/stable/nn.html?highlight=nn module#torch.nn.Module

class  torch.nn.Parameter

是Tensor的子类,被认为是一个模块的参数

Parameters 是Tendor的子类,当与模块一起用时有特殊的属性

--当它们被分配为一个模块的属性时,它们会被自动加入到模块的参数列表中

--------------------------------------------------------------------

class torch.nn.Module

是所有神经网络模块的基类.你自己构建的神经网络也需要是该模块的子类.

该模块内部也可以添加其它模块,允许其它模块以树的形式构建,你可以将这些子模块看作正常的类属性.

apply(function)

对网络结构的每个子模块分别调用一次该函数.典型的应用是初始化模型参数.

forward(*input)

定义来执行计算的每一次的流程.

该函数需要被自己的网络结构覆盖,即必须要自己重新定义.

zero_grad()

将model中的所有参数的梯度置为0