|
|
|
@ -16,13 +16,13 @@ class TestNet(unittest.TestCase):
|
|
|
|
|
net.complete_add_op(True)
|
|
|
|
|
|
|
|
|
|
expected = '''
|
|
|
|
|
Op(plain_net), inputs:(@EMPTY@, X, Y, w), outputs:(@TEMP@fc@0, Out, fc.out).
|
|
|
|
|
Op(add_two), inputs:(X, Y), outputs:(Out).
|
|
|
|
|
Op(plain_net), inputs:(@EMPTY@, X, w), outputs:(@TEMP@fc@0, fc.out).
|
|
|
|
|
Op(plain_net), inputs:(@EMPTY@, X, Y, w), outputs:(@TEMP@fc@0, Out, fc.out).
|
|
|
|
|
Op(add_two), inputs:(X, Y), outputs:(Out).
|
|
|
|
|
Op(plain_net), inputs:(@EMPTY@, X, w), outputs:(@TEMP@fc@0, fc.out).
|
|
|
|
|
Op(fc), inputs:(X, w, @EMPTY@), outputs:(fc.out, @TEMP@fc@0).
|
|
|
|
|
Op(mul), inputs:(X, w), outputs:(@TEMP@fc@0).
|
|
|
|
|
Op(sigmoid), inputs:(@TEMP@fc@0), outputs:(fc.out).
|
|
|
|
|
'''
|
|
|
|
|
Op(mul), inputs:(X, w), outputs:(@TEMP@fc@0).
|
|
|
|
|
Op(sigmoid), inputs:(@TEMP@fc@0), outputs:(fc.out).
|
|
|
|
|
'''
|
|
|
|
|
self.assertEqual(expected, "\n" + str(net))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|