提交 f9902a47 authored 作者: Frederic's avatar Frederic

pep8, remove print

上级 f05af839
...@@ -2343,6 +2343,7 @@ def test_batched_dot(): ...@@ -2343,6 +2343,7 @@ def test_batched_dot():
assert result.shape[0] == first_mat_val.shape[0] assert result.shape[0] == first_mat_val.shape[0]
def test_batched_tensordot(): def test_batched_tensordot():
first = theano.tensor.tensor4("first") first = theano.tensor.tensor4("first")
second = theano.tensor.tensor4("second") second = theano.tensor.tensor4("second")
...@@ -2364,10 +2365,10 @@ def test_batched_tensordot(): ...@@ -2364,10 +2365,10 @@ def test_batched_tensordot():
second_mat_val = numpy.random.rand(10, 4).astype(config.floatX) second_mat_val = numpy.random.rand(10, 4).astype(config.floatX)
result_fn = theano.function([first_mat, second_mat], output) result_fn = theano.function([first_mat, second_mat], output)
result = result_fn(first_mat_val, second_mat_val) result = result_fn(first_mat_val, second_mat_val)
print(result.shape)
assert result.shape[0] == first_mat_val.shape[0] assert result.shape[0] == first_mat_val.shape[0]
assert len(result.shape) == 1 assert len(result.shape) == 1
def test_tensor_values_eq_approx(): def test_tensor_values_eq_approx():
#test, inf, -inf and nan equal themself #test, inf, -inf and nan equal themself
a = numpy.asarray([-numpy.inf, -1, 0, 1, numpy.inf, numpy.nan]) a = numpy.asarray([-numpy.inf, -1, 0, 1, numpy.inf, numpy.nan])
...@@ -3145,15 +3146,12 @@ class T_Join_and_Split(unittest.TestCase): ...@@ -3145,15 +3146,12 @@ class T_Join_and_Split(unittest.TestCase):
b_v = numpy.random.rand(4) b_v = numpy.random.rand(4)
f = theano.function([a, b], [Ha, Hb]) f = theano.function([a, b], [Ha, Hb])
Ha_v, Hb_v = f(a_v, b_v) Ha_v, Hb_v = f(a_v, b_v)
print Ha_v
print Hb_v
# The Hessian is always a matrix full of 0 # The Hessian is always a matrix full of 0
assert Ha_v.shape == (4, 4) assert Ha_v.shape == (4, 4)
assert Hb_v.shape == (4, 4) assert Hb_v.shape == (4, 4)
assert numpy.allclose(Ha_v, 0.) assert numpy.allclose(Ha_v, 0.)
assert numpy.allclose(Hb_v, 0.) assert numpy.allclose(Hb_v, 0.)
def test_join_concatenate_one_element(self): def test_join_concatenate_one_element(self):
''' Fast test of concatenate as this is an alias for join. ''' Fast test of concatenate as this is an alias for join.
also test that we remove the Join op if there is only 1 input''' also test that we remove the Join op if there is only 1 input'''
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论