code: def test_batch_to_space_nd(self):
      x = random_ops.random_uniform([7, 5 * 2 * 3, 2, 2, 3, 2])
      block_shapes = [2, 3]
      crops = [[1, 2], [1, 0]]
  
      def loop_fn(i):
          x1 = array_ops.gather(x, i)
          return array_ops.batch_to_space_nd(x1, block_shapes, crops)
  
      self._test_loop_fn(loop_fn, 7)
  
imports: from tensorflow.python.eager import backprop
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import test_util
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import array_ops_stack
  from tensorflow.python.ops import math_ops
  from tensorflow.python.ops import nn
  from tensorflow.python.ops import random_ops
  from tensorflow.python.ops import tensor_array_grad
  from tensorflow.python.ops.parallel_for import control_flow_ops
  from tensorflow.python.ops.parallel_for.test_util import PForTestCase
  from tensorflow.python.platform import test
label_desc: Use image node_id 43 for calling the ArrayTest obj's underlying member method code with example usage: obj.test_batch_to_space_nd() and returns: array_ops