提交 db87e189 authored 作者: Iulian Vlad Serban's avatar Iulian Vlad Serban

Fixed Flake8 errors.

上级 7c52a043
...@@ -3495,16 +3495,16 @@ def test_local_elemwise_sub_zeros(): ...@@ -3495,16 +3495,16 @@ def test_local_elemwise_sub_zeros():
mat_val = rng.rand(3, 2) mat_val = rng.rand(3, 2)
mode = theano.compile.get_default_mode()\ mode = theano.compile.get_default_mode()\
.excluding('canonicalize', 'uncanonicalize',\ .excluding('canonicalize', 'uncanonicalize',
'ShapeOpt', 'local_fill_to_alloc',\ 'ShapeOpt', 'local_fill_to_alloc',
'local_elemwise_alloc')\ 'local_elemwise_alloc')\
.including('local_elemwise_sub_zeros') .including('local_elemwise_sub_zeros')
# Test scalar minus scalar # Test scalar minus scalar
f = function([scalar], scalar-scalar, mode=mode) f = function([scalar], scalar - scalar, mode=mode)
# Check optimized graph is correct # Check optimized graph is correct
assert isinstance(f.maker.fgraph.toposort()[0].op, T.Elemwise) assert isinstance(f.maker.fgraph.toposort()[0].op, T.Elemwise)
assert isinstance(f.maker.fgraph.toposort()[0].inputs[1],\ assert isinstance(f.maker.fgraph.toposort()[0].inputs[1],
T.TensorConstant) or\ T.TensorConstant) or\
isinstance(f.maker.fgraph.toposort()[0].inputs[1],\ isinstance(f.maker.fgraph.toposort()[0].inputs[1],\
T.TensorConstant) T.TensorConstant)
...@@ -3513,10 +3513,10 @@ def test_local_elemwise_sub_zeros(): ...@@ -3513,10 +3513,10 @@ def test_local_elemwise_sub_zeros():
assert check_stack_trace(f, ops_to_check='all') assert check_stack_trace(f, ops_to_check='all')
# Test vector minus vector # Test vector minus vector
f = function([vect], vect-vect, mode=mode) f = function([vect], vect - vect, mode=mode)
# Check optimized graph is correct # Check optimized graph is correct
assert isinstance(f.maker.fgraph.toposort()[0].op, T.Elemwise) assert isinstance(f.maker.fgraph.toposort()[0].op, T.Elemwise)
assert isinstance(f.maker.fgraph.toposort()[0].inputs[1],\ assert isinstance(f.maker.fgraph.toposort()[0].inputs[1],
T.TensorConstant) or\ T.TensorConstant) or\
isinstance(f.maker.fgraph.toposort()[0].inputs[1],\ isinstance(f.maker.fgraph.toposort()[0].inputs[1],\
T.TensorConstant) T.TensorConstant)
...@@ -3525,10 +3525,10 @@ def test_local_elemwise_sub_zeros(): ...@@ -3525,10 +3525,10 @@ def test_local_elemwise_sub_zeros():
assert check_stack_trace(f, ops_to_check='all') assert check_stack_trace(f, ops_to_check='all')
# Test vector minus vector # Test vector minus vector
f = function([mat], mat-mat, mode=mode) f = function([mat], mat - mat, mode=mode)
# Check optimized graph is correct # Check optimized graph is correct
assert isinstance(f.maker.fgraph.toposort()[0].op, T.Elemwise) assert isinstance(f.maker.fgraph.toposort()[0].op, T.Elemwise)
assert isinstance(f.maker.fgraph.toposort()[0].inputs[1],\ assert isinstance(f.maker.fgraph.toposort()[0].inputs[1],
T.TensorConstant) or\ T.TensorConstant) or\
isinstance(f.maker.fgraph.toposort()[0].inputs[1],\ isinstance(f.maker.fgraph.toposort()[0].inputs[1],\
T.TensorConstant) T.TensorConstant)
...@@ -5666,20 +5666,20 @@ class T_local_sum_prod(unittest.TestCase): ...@@ -5666,20 +5666,20 @@ class T_local_sum_prod(unittest.TestCase):
mat = T.dmatrix() mat = T.dmatrix()
scalar = T.dscalar() scalar = T.dscalar()
f = theano.function([vect, scalar], T.sum(vect*scalar), mode=m0) f = theano.function([vect, scalar], T.sum(vect * scalar), mode=m0)
assert check_stack_trace(f, ops_to_check='all') assert check_stack_trace(f, ops_to_check='all')
f = theano.function([vect], T.sum(-vect), mode=m0) f = theano.function([vect], T.sum(-vect), mode=m0)
assert check_stack_trace(f, ops_to_check=[T.Sum]) assert check_stack_trace(f, ops_to_check=[T.Sum])
f = theano.function([vect, scalar], f = theano.function([vect, scalar],
T.elemwise.Prod()(vect*scalar), mode=m0) T.elemwise.Prod()(vect * scalar), mode=m0)
assert check_stack_trace(f, ops_to_check=[T.elemwise.Prod]) assert check_stack_trace(f, ops_to_check=[T.elemwise.Prod])
f = theano.function([vect], T.elemwise.Prod()(-vect), mode=m0) f = theano.function([vect], T.elemwise.Prod()(-vect), mode=m0)
assert check_stack_trace(f, ops_to_check=[T.elemwise.Prod]) assert check_stack_trace(f, ops_to_check=[T.elemwise.Prod])
f = theano.function([mat, scalar], T.sum(mat*scalar), mode=m0) f = theano.function([mat, scalar], T.sum(mat * scalar), mode=m0)
assert check_stack_trace(f, ops_to_check='all') assert check_stack_trace(f, ops_to_check='all')
f = theano.function([mat], T.sum(-mat), mode=m0) f = theano.function([mat], T.sum(-mat), mode=m0)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论