"""
    从基类 nn.Module 中继承，自定义神经网络
"""
import torch
from torch import nn


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

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


# 实例化网络
myNn = MyNN()

# 初始化输入：tensor格式
x = torch.tensor(1.0)

# 神经网络计算，经由正向过程forward
output = myNn(x)

print(output)