Pytorch学习日记——nn.Module的使用

学习视频——B站【小土堆】

代码

import torch
from torch import nn


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

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

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

定义一个名为Tudui的神经网络

结果：