B站小土堆Pytorch视频的学习-神经网络的基本骨架-nn.Module的使用（2023.06.14）

一.大概流程

二.forward函数自带操作

nn的全称是Neural Networks，一般来说都要重写此方法。在 PyTorch 中，神经网络的建模通常是通过继承 nn.Module 类来实现的。在 nn.Module 的子类中，可以定义一些层和其他模块，然后在 forward 方法中将它们组合起来。

三.代码

call函数里写的有调用forward函数，所以tudui(x)可以直接tudui(x)->call(x)->forward(x)

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

    def forward(self,input):
        output=input+1
        return output
tudui=Tudui()
x=torch.tensor(1.0)
output=tudui(x)
print(output)