合集请看：pyTorch入门合集
参考视频：https://www.bilibili.com/video/BV1hE411t7RN/?spm_id_from=333.337.search-card.all.click

是pytorch提供的神经网络的基础模型，之后的所有操作都在这个module上扩展

import torch  
from torch import nn  

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

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

module = Module()  
x = torch.tensor(1.0)  
output = module(x)  
print(output)

其中init和forward函数必有，forward函数对应神经网络中的向前运算