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

Merge pull request #3526 from cooijmanstim/nanguardmode_max_empty

do not call max on variable with no elements
......@@ -265,12 +265,12 @@ class NanGuardMode(Mode):
err = False
if isinstance(var, theano.gof.type.CDataType._cdata_type):
err = False
elif cuda.cuda_available and isinstance(var, cuda.CudaNdarray):
err = (f_gpuabsmax(var.reshape(var.size)) > 1e10)
elif isinstance(var, np.random.mtrand.RandomState):
err = False
elif var.size == 0:
err = False
elif cuda.cuda_available and isinstance(var, cuda.CudaNdarray):
err = (f_gpuabsmax(var.reshape(var.size)) > 1e10)
else:
err = (np.abs(var).max() > 1e10)
if err:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论