制作作品的软件_武汉宣传片拍摄制作公司_厦门专业做优化的公司_山东济南seo整站优化费用

视频地址神经网络的基本骨架-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)