建立简单的神经网络

from torch import nn

import torch.nn.functional as f

class my(nn.Module):
def init(self):
super(my, self).init()
my.a=nn.Linear(3,3)
my.b=nn.Linear(3,1)

def forward(self, x):
    x=f.relu(my.a(x))
    x=f.relu(my.b(x))
    return x

这是一个简单的过程，其中需要的主要就是init和forward,在init里面决定了要继承什么类，而forward里面决定了这个神经网络的形状——也就是积木是怎么搭建的。

在搭建的时候需要明确的是必须在init里面必须说明要搭建什么，也就是必须说明要几层的网络，而在下面，要以x作为【流向】，确定x是怎么流过去的。