Beispiel #1
0
		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
Beispiel #2
0
 def forward(self, input):
     return ops.self_define_torch(input)