code: def _select_and_scatter_shape_rule(
      operand,
      source,
      init_value,
      *,
      select_jaxpr,
      select_consts,
      scatter_jaxpr,
      scatter_consts,
      window_dimensions,
      window_strides,
      padding,
  ):
      lax._check_shapelike(
          "select_and_scatter", "window_dimensions", window_dimensions
      )
      lax._check_shapelike(
          "select_and_scatter", "window_strides", window_strides
      )
      if len(window_dimensions) != len(window_strides):
          msg = (
              "select_and_scatter got inconsistent window_strides and "
              "window_dimensions: got window_strides {} and window_dimensions {}."
          )
          raise TypeError(msg.format(window_strides, window_dimensions))
      return operand.shape
  
imports: from __future__ import annotations
  from collections.abc import Sequence
  from functools import partial
  from typing import Callable
  import warnings
  import numpy
  from jax import tree_util
  from jax._src import ad_util
  from jax._src import core
  from jax._src import dispatch
  from jax._src import dtypes
  from jax._src import util
  from jax._src.core import ShapedArray, ConcreteArray
  from jax._src.interpreters import ad
  from jax._src.interpreters import batching
  from jax._src.interpreters import mlir
  from jax._src.lax import lax
  from jax._src.lax import convolution
  from jax._src.lax import slicing
  from jax._src.lib.mlir import ir
  from jax._src.lib.mlir.dialects import hlo
  from jax._src.numpy.ufuncs import logaddexp
  from jax._src.typing import Array
label_desc: Use image node_id 21 for calling a global function with example usage: _select_and_scatter_shape_rule(operand, source, init_value) and returns: operand