def forward(self, x, *args, **kwargs):
x = F.relu(self.affine1(x))
return x
微信
公众号
扫码关注公众号
微信
公众号