输入送到神经网络,经过forward处理,得到输出
import torch
from torch import nn
class Model (nn.Module):#Model继承Module
def __init__(self):
super().__init__()
#forward里面进行的运算就是对输入加1
def forward(self,input): #self就是这个类Model,input输入
output = input+1
return output
myModel = Model() #创建一个神经网络,名字叫做myModel,就是拿Model创建的
x = torch.tensor(1.0) #输入为x为1.0
output = myModel(x) #将x输入到创建的神经网络myModel中
print(output)
结果
tensor(2.)