代码实战（小土堆）——神经网络的基本骨架—nn.Module的使用

一个简单的神经网络

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()
# 神经网络的输入
x = torch.tensor(1.0)
# 把输入x放到神经网络当中
output = tudui(x)

print(output)