jax.dlpack.is_supported_dtype

jax.dlpack.is_supported_dtype(dtype)

检查 dtype 是否受 jax.dlpack 支持。

参数:

dtype (DTypeLike)

返回类型:

bool