pytorch入门9：学会nn.Module的使用

"""
本节主要学会nn.Module的使用
"""

import torch
from torch import nn
from torch.autograd.grad_mode import F


class Zkl(nn.Module):
    def __init__(self):
        super().__init__()

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

zkl = Zkl()
c=torch.Tensor(1)
b=zkl(c)
print(b)