提交 c4faf858 authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Update comment, indentation.

上级 4ea37b8d
...@@ -1307,7 +1307,9 @@ class Scan(PureOp): ...@@ -1307,7 +1307,9 @@ class Scan(PureOp):
known_grads={y: g_y}, wrt=x) known_grads={y: g_y}, wrt=x)
except gradient.NullTypeGradError: except gradient.NullTypeGradError:
# It means the gradient is undefined (which implies # It means the gradient is undefined (which implies
# is connected) # is connected).
# Warning: x is not the right gradient here, but the only
# thing we will check later is whether it is None.
gmp[x] = x gmp[x] = x
except gradient.DisconnectedInputError: except gradient.DisconnectedInputError:
gmp[x] = None gmp[x] = None
...@@ -1904,6 +1906,7 @@ class Scan(PureOp): ...@@ -1904,6 +1906,7 @@ class Scan(PureOp):
inner_gfn_outs = (inner_out_mitmot + inner_gfn_outs = (inner_out_mitmot +
inner_out_sitsot + inner_out_sitsot +
inner_out_nitsot) inner_out_nitsot)
local_op = Scan(inner_gfn_ins, inner_gfn_outs, info) local_op = Scan(inner_gfn_ins, inner_gfn_outs, info)
outputs = local_op(*outer_inputs) outputs = local_op(*outer_inputs)
if type(outputs) not in (list, tuple): if type(outputs) not in (list, tuple):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论