code: def testLargeTwoDimensional(self):
      rows = 100000
      cols = 100
      data_list = [None] * rows
      for i in range(rows):
          data_list[i] = [i for _ in range(cols)]
      num_partitions = 97
      indices_list = [(i**2) % num_partitions for i in range(rows)]
      parts = [[] for _ in range(num_partitions)]
      for i in range(rows):
          parts[(i**2) % num_partitions].append(data_list[i])
      with self.session():
          data = constant_op.constant(data_list, dtype=dtypes.float32)
          indices = constant_op.constant(
              indices_list, dtype=dtypes.int32
          )
          partitions = data_flow_ops.dynamic_partition(
              data, indices, num_partitions=num_partitions
          )
          partition_vals = self.evaluate(partitions)
  
      self.assertEqual(num_partitions, len(partition_vals))
      for i in range(num_partitions):
          # reshape because of empty parts
          parts_np = np.array(parts[i], dtype=np.float64).reshape(
              -1, cols
          )
          self.assertAllEqual(parts_np, partition_vals[i])
  
imports: import unittest
  from absl.testing import parameterized
  import numpy
  from tensorflow.python.framework import config
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import ops
  from tensorflow.python.framework import test_util
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import data_flow_ops
  from tensorflow.python.ops import gradients_impl
  import tensorflow.python.ops.data_flow_grad
  from tensorflow.python.platform import test
label_desc: Use image node_id 4 for calling the DynamicPartitionTest obj's underlying member method code with example usage: obj.testLargeTwoDimensional() without return types