神经网络的基本骨架nn.Module的使用
import torchfrom torch import nnclass NnModule(nn.Module):def __init__(self):super().__init__()def forward(self,input):output=input+1return outputnnmodule=NnModule()# torch.tensor是一个包含多个同类数据类型数据的多维矩阵。
·
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)
技术共进，成长同行——讯飞AI开发者社区
更多推荐
0
0- 0
已为社区贡献2条内容
所有评论(0)