提交 b65abe5c authored 作者: Amjad Almahairi's avatar Amjad Almahairi

fix test that is causing failure

上级 1a067453
......@@ -4576,6 +4576,11 @@ class test_local_remove_switch_const_cond(unittest.TestCase):
assert f0(vx) == 0
assert f1(vx) == 5
z2 = theano.tensor.switch(0, 0, x.shape[1])
f2 = theano.function([x], z2, mode=self.mode)
assert isinstance(f2.maker.fgraph.toposort()[0].op, Shape_i)
assert f2(vx) == 5
def test_broadcast1(self):
# test switch(cst, matrix, row)
x = theano.tensor.matrix('x', dtype='int32')
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论