code: def where(condition, x1, x2, /):
      """Returns elements chosen from x1 or x2 depending on condition."""
      dtype = _result_type(x1, x2)
      return jax.numpy.where(
          condition, x1.astype(dtype), x2.astype(dtype)
      )
  
imports: import jax
  from jax.experimental.array_api._data_type_functions import result_type
label_desc: Use image node_id 4 for calling a global function with example usage: where() and returns: jax