jax.dtypes.scalar_type_of

Contents

jax.dtypes.scalar_type_of#

jax.dtypes.scalar_type_of(x)#

Return the scalar type associated with a JAX value.

Parameters:

x (Any) –

Return type:

type