def forward(self,x): x = F.max_pool2d(ops.self_define_torch(self.cov1(x)),(2,2)) x = F.max_pool2d(ops.self_define_torch(self.cov2(x)),(2,2)) x = x.view(-1,16*4*4) x = ops.self_define_torch(self.fc1(x)) x = ops.self_define_torch(self.fc2(x)) x = self.fc3(x) return x
def forward(self, input): return ops.self_define_torch(input)