神经网络的基本骨架——nn.Module的使用

torch.nn：

Containers 骨架：Moudule

Convolution Layers 卷积层

Pooling Layers 池化层

浅写一个自己的神经网络：其中自定义网络名称为FirstModule，添加了init方法以及重写了forward函数，只实现了x加1的功能

import torch
from torch import nn


class FirstModule(nn.Module):
    def __init__(self) -> None:
        super().__init__()

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



firstmodule = FirstModule()
x = torch.tensor(1.0)
output = firstmodule(x)
print(output)