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

fixed tensorfromscalar.grad's handling of integer inputs

上级 28754534
......@@ -1912,6 +1912,8 @@ class TensorFromScalar(Op):
s, = inp
dt, = grads
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)]
def __str__(self):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论