jax.numpy.isdtype

内容

jax.numpy.isdtype#

jax.numpy.isdtype(dtype, kind)[source]#

返回一个布尔值，指示提供的 dtype 是否属于指定的种类。

参数：

  • dtype (DTypeLike) – 输入 dtype

  • kind (str | DTypeLike | tuple[str | DTypeLike, ...]) –

    数据类型种类。如果 kind 是 dtype-like，则返回 dtype = kind。如果 kind 是字符串，则如果 dtype 属于指定的类别，则返回 True

    • 'bool': {bool}

    • 'signed integer': {int4, int8, int16, int32, int64}

    • “无符号整数”：{uint4, uint8, uint16, uint32, uint64}

    • “整数”：('signed integer', 'unsigned integer')的简写

    • “实浮点数”：{float8_*, float16, bfloat16, float32, float64}

    • “复数浮点数”：{complex64, complex128}

    • “数值”：('integral', 'real floating', 'complex floating')的简写

    如果kind是一个元组，则如果数据类型与元组中的任何条目匹配，则返回True。

返回值：

True 或 False

返回类型：

bool

内容