cupy.testing.for_float_dtypes#

cupy.testing.for_float_dtypes(name='dtype', no_float16=False)[source]#

用于测试浮点数据类型的装饰器。

参数:
  • name (str) – 用于传递指定数据类型的参数名。

  • no_float16 (bool) – 如果为 True,则从候选数据类型中排除 numpy.float16

待测试的数据类型包括 numpy.float16(可选)、numpy.float32numpy.float64