土堆学习笔记——P16神经网络的搭建

nn表示Neural Network缩写

一、Container

container里面的module是一个原型（方法？），我们通过修改里面的函数细节，实现自己想要的功能

forward前馈神经网络。
输入->forward处理->输出

代码如下

import torch
from torch import nn

class Tudui(nn.Module):
    def __init__(self):
        super().__init__()

    def forward(self, input):
        output = input +1
        return output
    

tudui = Tudui() #tudui就是拿这个模板创建出来的神经网络
x = torch.tensor(1.0)
output = tudui(x) #对x进行forward操作
print(output)
#输出tensor(2.)