提交 50ef355b authored 作者: Frederic Bastien's avatar Frederic Bastien

white space fix.

上级 3704fc21
...@@ -46,14 +46,14 @@ def test_softmax_optimizations(): ...@@ -46,14 +46,14 @@ def test_softmax_optimizations():
x = tensor.fmatrix('x') x = tensor.fmatrix('x')
one_of_n = tensor.lvector('one_of_n') one_of_n = tensor.lvector('one_of_n')
op = crossentropy_categorical_1hot op = crossentropy_categorical_1hot
xe = op(x, one_of_n) xe = op(x, one_of_n)
env = theano.gof.Env( env = theano.gof.Env(
[x, one_of_n], [x, one_of_n],
[op(softmax(x), one_of_n)]) [op(softmax(x), one_of_n)])
assert env.outputs[0].owner.op == op assert env.outputs[0].owner.op == op
mode_with_gpu.optimizer.optimize(env) mode_with_gpu.optimizer.optimize(env)
assert str(env.outputs[0].owner.op) == 'OutputGuard' assert str(env.outputs[0].owner.op) == 'OutputGuard'
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论