提交 3c02813f authored 作者: Frederic's avatar Frederic

pep8

上级 d82eb54a
...@@ -6,6 +6,7 @@ import theano ...@@ -6,6 +6,7 @@ import theano
import theano.tensor as T import theano.tensor as T
import StringIO import StringIO
def test_profiling(): def test_profiling():
old1 = theano.config.profile old1 = theano.config.profile
...@@ -18,7 +19,7 @@ def test_profiling(): ...@@ -18,7 +19,7 @@ def test_profiling():
y = T.dvector("y") y = T.dvector("y")
z = x + y z = x + y
f = theano.function([x, y], z, profile=True, name="test_profiling") f = theano.function([x, y], z, profile=True, name="test_profiling")
output = f([1, 2, 3, 4],[1, 1, 1, 1]) output = f([1, 2, 3, 4], [1, 1, 1, 1])
buf = StringIO.StringIO() buf = StringIO.StringIO()
f.profile.summary(buf) f.profile.summary(buf)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论