提交 6010cadd authored 作者: Caglar's avatar Caglar

fixed the undefined variable problem.

上级 8f442150
...@@ -52,10 +52,8 @@ class MultinomialFromUniform(Op): ...@@ -52,10 +52,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
if x.dtype in T.discrete_dtypes: return [T.zeros_like(x, dtype=theano.config.floatX) if x.dtype in
return [T.zeros_like(x, dtype=theano.config.floatX) for x in ins] T.discrete_dtypes else T.zeros_like(x) for x in ins]
else:
return [T.zeros_like(x) for x in ins]
def c_code_cache_version(self): def c_code_cache_version(self):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论