jax.dtypes.scalar_type_of# jax.dtypes.scalar_type_of(x)[source]# 返回与 JAX 值关联的标量类型。 参数: x (Any) 返回类型: type
