code: def _reshape(a: Array, *args: Any, order: str = "C") -> Array:
      """Returns an array containing the same data with a new shape.
  
      Refer to :func:`jax.numpy.reshape` for full documentation.
      """
      __tracebackhide__ = True
      newshape = _compute_newshape(
          a, args[0] if len(args) == 1 else args
      )
      if order == "C":
          return lax.reshape(a, newshape, None)
      elif order == "F":
          dims = list(range(a.ndim)[::-1])
          return lax.reshape(a, newshape[::-1], dims).T
      elif order == "A":
          raise NotImplementedError(
              "np.reshape order=A is not implemented."
          )
      else:
          raise ValueError(
              f"Unexpected value for 'order' argument: {order}."
          )
  
imports: from __future__ import annotations
  import abc
  from functools import partial, wraps
  import math
  from typing import Any
  import numpy
  import jax
  from jax import lax
  from jax._src import core
  from jax._src import dtypes
  from jax._src.api_util import _ensure_index_tuple
  from jax._src.array import ArrayImpl
  from jax._src.lax import lax
  from jax._src.numpy import lax_numpy
  from jax._src.numpy import reductions
  from jax._src.numpy import ufuncs
  from jax._src.numpy import util
  from jax._src.ops import scatter
  from jax._src.typing import Array, ArrayLike, DimSize, DTypeLike, Shape
  from jax._src.util import safe_zip, safe_map
label_desc: Use image node_id 8 for calling a global function with example usage: _reshape(a) and returns: lax, lax