提交 5211d560 authored 作者: Frederic Bastien's avatar Frederic Bastien

Simplify and possibly speed up computation of Subtensor.grad

上级 4e08a226
...@@ -573,11 +573,12 @@ class Subtensor(Op): ...@@ -573,11 +573,12 @@ class Subtensor(Op):
gz, = grads gz, = grads
x = inputs[0] x = inputs[0]
rest = inputs[1:] rest = inputs[1:]
output = self(*inputs) if x.dtype.find('int') != -1:
if output.dtype.find('int') != -1:
first = x.zeros_like().astype(theano.config.floatX) first = x.zeros_like().astype(theano.config.floatX)
else: else:
first = IncSubtensor(self.idx_list)(x.zeros_like(), gz, *rest) first = IncSubtensor(self.idx_list,
set_instead_of_inc=True)(x.zeros_like(),
gz, *rest)
return ([first] + [DisconnectedType()()] * len(rest)) return ([first] + [DisconnectedType()()] * len(rest))
def connection_pattern(self, node): def connection_pattern(self, node):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论