code: def batch_dims(self):
      return tuple(
          x
          for x in range(len(self.out_aval.shape))
          if x not in self.dnums.offset_dims
      )
  
imports: from __future__ import annotations
  import builtins
  from collections.abc import Sequence
  import dataclasses
  from functools import partial, wraps
  import math
  import string
  from typing import Any, Callable, Optional
  from jax._src import core
  from jax import lax
  from jax._src.lax import slicing
  from jax._src import dtypes
  from jax._src import util
  from jax.experimental.jax2tf import jax2tf
  import numpy
  import tensorflow
label_desc: Use image node_id 3 for calling the GatherArgs obj's underlying member method code with example usage: obj.batch_dims() and returns: tuple