nn.Module

• 继承nn.Module，输入数值，经过forward输出
• 示例代码：

import torch
from torch import nn
class FirstTime(nn.Module):
    def __init__(self):
        super().__init__()
    def forward(self,input):
        output = input + 1
        return output
first = FirstTime()
input = torch.tensor(1.0)
output = first(input)
print(output)