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

fixed tensorfromscalar.grad's handling of integer inputs

上级 28754534
...@@ -1912,6 +1912,8 @@ class TensorFromScalar(Op): ...@@ -1912,6 +1912,8 @@ class TensorFromScalar(Op):
s, = inp s, = inp
dt, = grads dt, = grads
assert dt.type.dtype.find('float') != -1 assert dt.type.dtype.find('float') != -1
if s.type.dtype.find('int') != -1:
return [s.zeros_like().astype(theano.config.floatX)]
return [scalar_from_tensor(dt)] return [scalar_from_tensor(dt)]
def __str__(self): def __str__(self):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论