提交 ece6d19d authored 作者: Frederic Bastien's avatar Frederic Bastien

make sure to compare str

上级 ffed817a
...@@ -141,7 +141,7 @@ def contains_inf(arr, node=None, var=None): ...@@ -141,7 +141,7 @@ def contains_inf(arr, node=None, var=None):
""" """
if not _is_numeric_value(arr, var): if not _is_numeric_value(arr, var):
return False return False
elif "int" in getattr(arr, 'dtype', ''): elif "int" in str(getattr(arr, 'dtype', '')):
return False return False
elif pygpu_available and isinstance(arr, GpuArray): elif pygpu_available and isinstance(arr, GpuArray):
return (np.isinf(f_gpua_min(arr.reshape(arr.size))) or return (np.isinf(f_gpua_min(arr.reshape(arr.size))) or
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论