提交 546cc762 authored 作者: Frederic's avatar Frederic

pep8

上级 f585ac8b
...@@ -281,14 +281,15 @@ def test_trace(): ...@@ -281,14 +281,15 @@ def test_trace():
ok = True ok = True
assert ok assert ok
def test_spectral_radius_bound(): def test_spectral_radius_bound():
tol = 10**(-6) tol = 10 ** (-6)
rng = numpy.random.RandomState(utt.fetch_seed()) rng = numpy.random.RandomState(utt.fetch_seed())
x = theano.tensor.matrix() x = theano.tensor.matrix()
radius_bound = spectral_radius_bound(x, 5) radius_bound = spectral_radius_bound(x, 5)
f = theano.function([x], radius_bound) f = theano.function([x], radius_bound)
shp = (3, 3) shp = (3, 4)
m = rng.rand(*shp).astype(config.floatX) m = rng.rand(*shp).astype(config.floatX)
m = numpy.cov(m).astype(config.floatX) m = numpy.cov(m).astype(config.floatX)
radius_bound_theano = f(m) radius_bound_theano = f(m)
...@@ -297,7 +298,7 @@ def test_spectral_radius_bound(): ...@@ -297,7 +298,7 @@ def test_spectral_radius_bound():
mm = m mm = m
for i in range(5): for i in range(5):
mm = numpy.dot(mm, mm) mm = numpy.dot(mm, mm)
radius_bound_numpy = numpy.trace(mm)**(2**(-5)) radius_bound_numpy = numpy.trace(mm) ** (2 ** (-5))
assert abs(radius_bound_numpy - radius_bound_theano) < tol assert abs(radius_bound_numpy - radius_bound_theano) < tol
# test the bound # test the bound
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论