code: def central_crop(inputs, target_shape):
      """Returns a central crop in axis (1, 2).
  
      Args:
        inputs: nd-array; Inputs in shape of `[bs, height, width, channels]'.
        target_shape: tuple(int); Target shape after crop.
  
      Returns:
        Cropped image.
      """
      h, w = target_shape[1:3]
      assert h <= inputs.shape[1], f"{h} > {inputs.shape[1]}"
      assert w <= inputs.shape[2], f"{w} > {inputs.shape[2]}"
      h0 = (inputs.shape[1] - h) // 2
      w0 = (inputs.shape[2] - w) // 2
      return inputs[:, h0 : (h0 + h), w0 : (w0 + w)]
  
file_path: scenic/scenic/model_lib/layers/nn_ops.py
imports: from typing import Callable, Sequence
  import flax.linen
  import jax
  from jax import lax
  import jax.numpy
  import numpy
label_desc: Use image node_id 10 for calling a global function with example usage: central_crop(inputs, target_shape) and returns: inputs