#!/usr/bin/python3
# -*- coding: utf-8 -*-
"""
@Time        : 2022/5/4 15:51
@Author      : Albert Darren
@Contact     : 2563491540@qq.com
@File        : nn_module.py
@Version     : Version 1.0.0
@Description : TODO
@Created By  : PyCharm
"""
import torch.nn as nn
from torch import tensor


class MyNet(nn.Module):

    def __init__(self):
        super().__init__()

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


my_net = MyNet()
input_tensor = tensor(1.24)
output_tensor = my_net(input_tensor)
print(output_tensor)