PyTorch快速入门教程【小土堆】之nn.Module的使用

视频地址神经网络的基本骨架-nn.Module的使用_哔哩哔哩_bilibili

nn为神经网路Neural Network的缩写，本次实现了一个最简单的把输入值加1的模型

import torch
from torch import nnclass Tudui(nn.Module): #自己写的模型也必须继承nn.module类def __init__(self):super().__init__() #继承了父类的初始化函数def forward(self, input): #前向传播output = input + 1return outputtudui = Tudui()
x = torch.tensor(1.0) #tensor类型的1赋值给x
output = tudui(x)
print(output)