code: def ndimarange(self, dims: Tuple[int, ...]):
      return np.arange(np.prod(dims), dtype=float).reshape(dims)
  
file_path: objax/tests/util_image.py
imports: import io
  import tempfile
  import unittest
  from typing import Tuple
  import jax.numpy
  import numpy
  from PIL import Image
  import objax
label_desc: Use image node_id 1 for calling the TestUtilImage obj's underlying member method code with example usage: obj.ndimarange(dims) and returns: np