提交 50e06772 authored 作者: Frédéric Bastien's avatar Frédéric Bastien

Merge pull request #4040 from caglar/fix_fload_multi_bug

Fixes the multinomial grad dtype bug
...@@ -53,7 +53,8 @@ class MultinomialFromUniform(Op): ...@@ -53,7 +53,8 @@ class MultinomialFromUniform(Op):
def grad(self, ins, outgrads): def grad(self, ins, outgrads):
pvals, unis, n = ins pvals, unis, n = ins
(gz,) = outgrads (gz,) = outgrads
return [T.zeros_like(x) for x in ins] return [T.zeros_like(x, dtype=theano.config.floatX) if x.dtype in
T.discrete_dtypes else T.zeros_like(x) for x in ins]
def c_code_cache_version(self): def c_code_cache_version(self):
return (8,) return (8,)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论