提交 c31c5313 authored 作者: Sina Honari's avatar Sina Honari 提交者: Pascal Lamblin

fixing bugs

上级 49079894
...@@ -824,7 +824,7 @@ class MRG_RandomStreams(object): ...@@ -824,7 +824,7 @@ class MRG_RandomStreams(object):
nstreams=None): nstreams=None):
# TODO : need description for method, parameter and return # TODO : need description for method, parameter and return
if n == 1: if n == 1:
p = undefined_grad(p) p = undefined_grad(as_tensor_variable(p))
x = self.uniform(size=size, nstreams=nstreams) x = self.uniform(size=size, nstreams=nstreams)
return cast(x < p, dtype) return cast(x < p, dtype)
else: else:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论