【土堆pytorch实战】P16 nn.Module

P16 nn.Module使用

import torch
from torch import nn
class Tudui(nn.Module):
    def __init__(self) -> None:
        super().__init__()
    def forward(self,input):
        output=input+1
        return output

tudui=Tudui()
x=torch.tensor(1.0)
output=tudui(x)
print(output)

简单的神经网络框架，实现了output=input+1，因此代码输出的output=tensor(2.)