code: def make_tensor_seed(seed):
      """Converts a seed to a `Tensor` seed."""
      if seed is None:
          raise ValueError("seed must not be None when using JAX")
      if hasattr(seed, "dtype") and jax.dtypes.issubdtype(
          seed.dtype, jax.dtypes.prng_key
      ):
          return seed
      return jnp.asarray(seed, jnp.uint32)
  
imports: import functools
  import jax
  from jax import lax
  from jax import random
  from jax import tree_util
  import jax.numpy
label_desc: Use image node_id 8 for calling a global function with example usage: make_tensor_seed(seed) and returns: jnp, seed