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) – 其值为 numpycupy 模块的参数名称。

  • type_check (bool) – 如果为 True,则也检查 dtype 的一致性。

  • accept_error (bool, Exception or tuple of Exception) – 指定可接受的错误。当 NumPy 和 CuPy 测试都引发相同类型的错误,且这些错误类型由此参数指定时,则忽略这些错误,不抛出。如果为 True,则所有错误类型都可接受。如果为 False,则任何错误都不可接受。

  • sp_name (str or None) – 其值为 scipy.sparsecupyx.scipy.sparse 模块的参数名称。如果为 None,则不为这些模块提供参数。

  • scipy_name (str or None) – 其值为 scipycupyx.scipy 模块的参数名称。如果为 None,则不为这些模块提供参数。

装饰的测试夹具需要返回在 cupy.testing.assert_array_almost_equal() 的意义上(除了数组模块的类型)相同的数组,即使 xpnumpycupy