神经网络中定义网络模型中的forward方法
神经网络的典型结构就是__init__()和forward()forward是自动调用的model = Net()y = model(x)如上则调用网络模型定义的forward方法。即,当把定义的网络模型model当作函数调用的时候就自动调用定义的网络模型的forward方法。是通过nn.Module 的__call__方法调用的...
·
神经网络的典型结构就是
__init__()
和
forward()
forward是自动调用的
model = Net()
y = model(x)
如上则调用网络模型定义的forward方法。即,当把定义的网络模型model当作函数调用的时候就自动调用定义的网络模型的forward方法。
是通过nn.Module 的__call__方法调用的
就相当于调用了模型就是直接调用它的forward函数,y=model(x),这个x就是直接传入到forward函数的x参数
更多推荐
所有评论(0)