提交 79b57320 authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Compare to None, instead of testing an array as a boolean

上级 5ee30fa1
......@@ -128,7 +128,7 @@ class MultinomialFromUniform(Op):
if unis.shape[0] != pvals.shape[0]:
raise ValueError("unis.shape[0] != pvals.shape[0]",
unis.shape[0], pvals.shape[0])
if not z[0] or z[0].shape != pvals.shape:
if z[0] is None or z[0].shape != pvals.shape:
z[0] = numpy.zeros(pvals.shape, dtype=node.outputs[0].dtype)
nb_multi = pvals.shape[0]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论