提交 93b208fc authored 作者: Pierre Luc Carrier's avatar Pierre Luc Carrier 提交者: --global

PEP8

上级 5d435e23
......@@ -47,9 +47,14 @@ class TestPdbBreakpoint:
grads = [T.grad(self.monitored_input1.sum(), self.input1),
T.grad(self.monitored_input2.sum(), self.input2)]
fct = theano.function([self.input1, self.input2], grads)
gradients = fct(input1_value, input2_value)
# Add self.monitored_input1 as an output to the Theano function to
# prevent Theano from optimizing the PdbBreakpoint op out of the
# function graph
fct = theano.function([self.input1, self.input2],
grads + [self.monitored_input1])
gradients = fct(input1_value, input2_value)[:-1]
expected_gradients = [numpy.ones((3, 3), dtype="float32"),
numpy.array(1., dtype="float32")]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论