pytorch里面一切自定义操作基本上都是继承nn.Module类来实现的
我们在定义自已的网络的时候,需要继承nn.Module类,并重新实现构造函数__init__构造函数和forward(前向传播)这两个方法。
Pytorch中对于一般的序列模型,直接使用torch.nn.Sequential类及可以实现,这点类似于keras,但是更多的时候面对复杂的模型,比如:多输入多输出、多分支模型、跨层连接模型、带有自定义层的模型等,就需要自己来定义一个模型了。
forward方法是必须要重写的,它是实现模型的功能,实现各个层之间的连接关系的核心。