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

pep8 and make test use floatX

上级 e2ee72a7
...@@ -3736,7 +3736,7 @@ class Test_lift_transpose_through_dot(unittest.TestCase): ...@@ -3736,7 +3736,7 @@ class Test_lift_transpose_through_dot(unittest.TestCase):
def test_local_upcast_elemwise_constant_inputs(): def test_local_upcast_elemwise_constant_inputs():
s = dvector("s") s = dvector("s")
x = tensor.sum(tensor.log(10**s)) x = tensor.sum(tensor.log(10 ** s))
f = function([s], [tensor.grad(x, s)]) f = function([s], [tensor.grad(x, s)])
f([-42, -2.1, -1, -0.5, 0, 0.2, 1, 2, 12]) f([-42, -2.1, -1, -0.5, 0, 0.2, 1, 2, 12])
...@@ -3748,22 +3748,22 @@ class TestShape_i(utt.InferShapeTester): ...@@ -3748,22 +3748,22 @@ class TestShape_i(utt.InferShapeTester):
def test_perform(self): def test_perform(self):
advec = dvector() advec = vector()
advec_val = numpy.random.rand(3) advec_val = numpy.random.rand(3).astype(config.floatX)
f = function([advec], Shape_i(0)(advec)) f = function([advec], Shape_i(0)(advec))
out = f(advec_val) out = f(advec_val)
assert numpy.allclose(out, advec_val.shape[0]) assert numpy.allclose(out, advec_val.shape[0])
admat = dmatrix() admat = matrix()
admat_val = numpy.random.rand(4, 3) admat_val = numpy.random.rand(4, 3).astype(config.floatX)
for i in xrange(2): for i in xrange(2):
f = function([admat], Shape_i(i)(admat)) f = function([admat], Shape_i(i)(admat))
out = f(admat_val) out = f(admat_val)
assert numpy.allclose(out, admat_val.shape[i]) assert numpy.allclose(out, admat_val.shape[i])
def test_infer_shape(self): def test_infer_shape(self):
admat = dmatrix() admat = matrix()
admat_val = numpy.random.rand(3, 4) admat_val = numpy.random.rand(3, 4).astype(config.floatX)
self._compile_and_check([admat], [Shape_i(0)(admat)], self._compile_and_check([admat], [Shape_i(0)(admat)],
[admat_val], Shape_i) [admat_val], Shape_i)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论