import torch
from torch import nn
class NnModule(nn.Module):
def __init__(self):
super().__init__()
def forward(self,input):
output=input+1
return output
nnmodule=NnModule()
# torch.tensor是一个包含多个同类数据类型数据的多维矩阵。
x=torch.tensor(1.0)
output=nnmodule(x) # 将x输入到神经网络中 调用forward函数
print(output)
神经网络的基本骨架nn.Module的使用
最新推荐文章于 2024-02-05 13:55:31 发布