与mpmath的numpy.allclose和multiprecision

Question

在我的python代码中，我定期使用numpy.allclose验证一些计算。另一方面，除了这些检查之外，实现还能够处理多精度（mpmath.mpc）数字。如果我想运行mpmath号码的验证码，我会得到类似

的内容

>>> check(H)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
File "module.py", line 19, in check_H
  assert(numpy.allclose(H, I, rtol=rtol, atol=atol))
File "/home/gereon/python/numpy/core/numeric.py", line 2025, in allclose
  xinf = isinf(x)
TypeError: ufunc 'isinf' not supported for the input types, and the inputs could
not be safely coerced to any supported types according to the casting rule ''safe''

检查两个多精度阵列是否足够相等的最佳方法是什么？

Answer 1

我查看了代码（numeric.py中的allclose）。它取决于isinf函数，它显然没有为mpmath实现。

但功能很简单。归结为：

def allclose(x, y, rtol=1.e-5, atol=1.e-8):
    return all(less_equal(abs(x-y), atol + rtol * abs(y)))

您可能必须使用mpmath等效项替换rtol和atol而不是浮点数。