4.1 nn.Module
torch.nn的核心数据结构是Module,它是一个抽象的概念,既可以表示神经网络的某个层,也可以包含很多层的神经网络。
注:实际使用中,最常见的做法是继承nn.Module,撰写自己的网络/层。
下面的示例代码实现了一个两层的感知机函数。
注:
需要把参数封装成parameter.
Module能够自检到自己的parameter,并将其作为学习参数。除了parameter, Module包含子Module,主Module能够递归查找子Module中的parameter.
注:构建某一个类时,可以包含已经创建的类。
import torch as t
from torch import nn
from torch.autograd import Variable as V
class Linear(nn.Module