code: def shaparange(s):
      return jn.arange(np.prod(s), dtype=float).reshape(s)
  
file_path: objax/tests/functional_pooling.py
imports: import unittest
  import jax.numpy
  import numpy
  import objax
label_desc: Use image node_id 1 for calling a global function with example usage: shaparange(s) and returns: jn