code: def _compatible_shapes(shape):
      if np.ndim(shape) == 0 or shape in scalar_shapes:
          return [shape]
      return (shape[n:] for n in range(len(shape) + 1))
  
imports: import collections
  from functools import partial
  import itertools
  import unittest
  from absl.testing import absltest
  from absl.testing import parameterized
  import numpy
  import jax
  from jax import numpy
  from jax._src import config
  from jax._src import dtypes
  from jax._src import test_util
  from jax._src.util import NumpyComplexWarning
label_desc: Use image node_id 3 for calling a global function with example usage: _compatible_shapes(shape) and returns: unknown, list