def __init__(self):
super(GeneralOptimizeRule, self).__init__()
self.register(RemoveLastSoftmax())
self.register(ConcatAffine())
self.register(ConcatScalarOperation())
self.register(RemoveUnnecessaryOperator())
self.register(ConcatZeroPadding())
def test_scalar_sub2():
x = Variable([5, 5], OrderNC)
h = fn(x)
y = h - 1
graph = Graph([x], [y])
flag_changed = True
while flag_changed:
graph, flag_changed = RemoveUnnecessaryOperator().optimize(graph)
ops = listup_operators(graph)
assert len(ops) == 2
def test_scalar_affine2():
x = Variable([5, 5], OrderNC)
h = fn(x)
y, = ScalarAffine(None, scale=2, bias=0)(h)
graph = Graph([x], [y])
flag_changed = True
while flag_changed:
graph, flag_changed = RemoveUnnecessaryOperator().optimize(graph)
ops = listup_operators(graph)
assert len(ops) == 2