提交 5a7081d7 authored 作者: Olivier Delalleau's avatar Olivier Delalleau

PEP8

上级 01e67c7f
...@@ -1387,6 +1387,7 @@ class Scan(PureOp): ...@@ -1387,6 +1387,7 @@ class Scan(PureOp):
self.inner_nitsot_outs(self_outputs)) self.inner_nitsot_outs(self_outputs))
scan_node = outs[0].owner scan_node = outs[0].owner
connection_pattern = self.connection_pattern(scan_node) connection_pattern = self.connection_pattern(scan_node)
def get_inp_idx(iidx): def get_inp_idx(iidx):
if iidx < self.n_seqs: if iidx < self.n_seqs:
return 1 + iidx return 1 + iidx
...@@ -1428,10 +1429,10 @@ class Scan(PureOp): ...@@ -1428,10 +1429,10 @@ class Scan(PureOp):
odx = get_out_idx(self_outputs.index(y)) odx = get_out_idx(self_outputs.index(y))
wrt = [x for x in theano.gof.graph.inputs([y]) wrt = [x for x in theano.gof.graph.inputs([y])
if (x in diff_inputs) and if (x in diff_inputs) and
(connection_pattern[get_inp_idx(self_inputs.index(x))][odx])] connection_pattern[get_inp_idx(self_inputs.index(x))][odx]]
grads = gradient.grad( grads = gradient.grad(
cost = None, cost=None,
known_grads = {y : g_y }, known_grads={y: g_y},
wrt=wrt, consider_constant=wrt, wrt=wrt, consider_constant=wrt,
disconnected_inputs='ignore', disconnected_inputs='ignore',
return_disconnected='None') return_disconnected='None')
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论