提交 b693f5e2 authored 作者: Ian Goodfellow's avatar Ian Goodfellow

fixed test broken by changed name of op

上级 fbe7e13e
...@@ -1500,7 +1500,7 @@ class test_local_subtensor_lift(unittest.TestCase): ...@@ -1500,7 +1500,7 @@ class test_local_subtensor_lift(unittest.TestCase):
def test7(self): def test7(self):
# test that Subtensor(Rebroadcast(x)) gets optimized into # test that Subtensor(Rebroadcast(x)) gets optimized into
# Rebroadcast(Subtensor(x)). # Rebroadcast(Subtensor(x)).
# test basic case # test basic case
x = tensor.matrix('x') x = tensor.matrix('x')
xval = numpy.random.rand(1,10).astype(config.floatX) xval = numpy.random.rand(1,10).astype(config.floatX)
...@@ -1526,7 +1526,7 @@ class test_local_subtensor_lift(unittest.TestCase): ...@@ -1526,7 +1526,7 @@ class test_local_subtensor_lift(unittest.TestCase):
assert isinstance(prog[0].op, tensor.Subtensor) assert isinstance(prog[0].op, tensor.Subtensor)
assert isinstance(prog[1].op, tensor.Rebroadcast) assert isinstance(prog[1].op, tensor.Rebroadcast)
assert (f2(yval) == yval[:,3,0,:]).all() assert (f2(yval) == yval[:,3,0,:]).all()
# corner case 2: subtensor idx_list is shorter than resulting broadcast pattern # corner case 2: subtensor idx_list is shorter than resulting broadcast pattern
f3 = function([y], newy[:,3,0], mode=mode_opt) f3 = function([y], newy[:,3,0], mode=mode_opt)
prog=f3.maker.env.toposort() prog=f3.maker.env.toposort()
...@@ -3244,7 +3244,7 @@ class Test_lift_transpose_through_dot(unittest.TestCase): ...@@ -3244,7 +3244,7 @@ class Test_lift_transpose_through_dot(unittest.TestCase):
def test_matrix_matrix(self): def test_matrix_matrix(self):
a, b = matrices('ab') a, b = matrices('ab')
g = self.simple_optimize(Env([a, b], [tensor.dot(a, b).T])) g = self.simple_optimize(Env([a, b], [tensor.dot(a, b).T]))
sg = '[dot(InplaceDimShuffle{1,0}(b), InplaceDimShuffle{1,0}(a))]' sg = '[dot(DimShuffle{1,0}(b), DimShuffle{1,0}(a))]'
assert str(g) == sg assert str(g) == sg
def test_row_matrix(self): def test_row_matrix(self):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论