code: def test_space_to_batch():
      # Without paddings
      verify_space_to_batch_nd([3, 3, 2, 1], [3], [0], [0])
      # With paddings
      verify_space_to_batch_nd([3, 3, 2, 1], [3], [1], [2])
      # Multiple spatial dims
      verify_space_to_batch_nd(
          [3, 3, 4, 5, 2], [3, 4, 2], [1, 0, 3], [2, 0, 0]
      )
      # No remaining dims
      verify_space_to_batch_nd(
          [3, 3, 4, 5, 2], [3, 4, 2, 2], [1, 4, 0, 0], [2, 0, 1, 0]
      )
  
file_path: tvm/tests/python/topi/test_topi_space_to_batch_nd.py
imports: import numpy
  import tvm
  from tvm import te
  from tvm import topi
  import tvm.testing
  import tvm.topi.testing
label_desc: Use image node_id 2 for calling a global function with example usage: test_space_to_batch() without return types