2021
02-26
02-26
pytorch 中forward 的用法与解释说明
前言最近在使用pytorch的时候,模型训练时,不需要使用forward,只要在实例化一个对象中传入对应的参数就可以自动调用forward函数即:forward的使用classModule(nn.Module):def__init__(self):super(Module,self).__init__()#......defforward(self,x):#......returnxdata=.....#输入数据#实例化一个对象module=Module()#前向传播module(data)#而不是使用下面的#module.forward(data)实际上modu...
继续阅读 >