分类：nn.Module

2020
09-25

pytorch 中的重要模块化接口nn.Module的使用

torch.nn是专门为神经网络设计的模块化接口，nn构建于autgrad之上，可以用来定义和运行神经网络nn.Module是nn中重要的类，包含网络各层的定义，以及forward方法查看源码初始化部分：def__init__(self):self._backend=thnn_backendself._parameters=OrderedDict()self._buffers=OrderedDict()self._backward_hooks=OrderedDict()self._forward_hooks=OrderedDict()self._forward_pre_hooks=OrderedDict(... 继续阅读 >