提交 2ea409ab authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Add "allow_remove_inf" keyword to CudaNdarrayType's values_eq_approx.

上级 4f773c52
......@@ -59,9 +59,10 @@ class CudaNdarrayType(Type):
return tensor.TensorType.values_eq(numpy.asarray(a), numpy.asarray(b))
@staticmethod
def values_eq_approx(a, b):
def values_eq_approx(a, b, allow_remove_inf=False):
#TODO: make the comparaison without transfert.
return tensor.TensorType.values_eq_approx(numpy.asarray(a), numpy.asarray(b))
return tensor.TensorType.values_eq_approx(numpy.asarray(a), numpy.asarray(b),
allow_remove_inf=allow_remove_inf)
def dtype_specs(self):
"""Return a tuple (python type, c type, numpy typenum) that corresponds to
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论