code: def one_hot(x, num_classes, *, dtype=None, axis=-1):
      dtype = ivy.float64 if dtype is None else ivy.as_ivy_dtype(dtype)
      return ivy.one_hot(x, num_classes, axis=axis, dtype=dtype)
  
imports: import ivy
  from ivy.functional.frontends.jax.func_wrapper import to_ivy_arrays_and_back
  from ivy.func_wrapper import with_supported_dtypes
label_desc: Use image node_id 21 for calling a global function with example usage: one_hot(x, num_classes) and returns: ivy