提交 dd153fc1 authored 作者: Ian Goodfellow's avatar Ian Goodfellow

Subtensor.grad integer handling

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