cupy.testing.numpy_cupy_array_almost_equal#
- cupy.testing.numpy_cupy_array_almost_equal(decimal=6, err_msg='', verbose=True, name='xp', type_check=True, accept_error=False, sp_name=None, scipy_name=None)[source]#
一个装饰器,用于检查 NumPy 和 CuPy 的结果是否几乎相等。
- 参数:
decimal (int) – 期望的精度。
err_msg (str) – 失败时打印的错误消息。
verbose (bool) – 如果为
True
,则将冲突值附加到错误消息中。name (str) – 其值为
numpy
或cupy
模块的参数名称。type_check (bool) – 如果为
True
,则也检查 dtype 的一致性。accept_error (bool, Exception or tuple of Exception) – 指定可接受的错误。当 NumPy 和 CuPy 测试都引发相同类型的错误,且这些错误类型由此参数指定时,则忽略这些错误,不抛出。如果为
True
,则所有错误类型都可接受。如果为False
,则任何错误都不可接受。sp_name (str or None) – 其值为
scipy.sparse
或cupyx.scipy.sparse
模块的参数名称。如果为None
,则不为这些模块提供参数。scipy_name (str or None) – 其值为
scipy
或cupyx.scipy
模块的参数名称。如果为None
,则不为这些模块提供参数。
装饰的测试夹具需要返回在
cupy.testing.assert_array_almost_equal()
的意义上(除了数组模块的类型)相同的数组,即使xp
是numpy
或cupy
。