code: def np_gather_ijk_index(arr, index):
      arr_flat = arr.reshape(-1, arr.shape[-1])
      _, j, k, _ = arr.shape
      index_transform = (
          index[:, 0] * j * k + index[:, 1] * k + index[:, 2]
      )
      return arr_flat[index_transform]
  
file_path: graphics/tensorflow_graphics/projects/local_implicit_grid/core/point_utils.py
imports: import numpy
  from plyfile import PlyData
  from plyfile import PlyElement
label_desc: Use image node_id 4 for calling a global function with example usage: np_gather_ijk_index(arr, index) and returns: arr_flat