code: def test_tensor_x_sharded(self):
      t00 = np.arange(8).reshape(2, 4)
      t01 = np.arange(8).reshape(2, 4)
      t10 = np.arange(8, 16).reshape(2, 4)
      t11 = np.arange(8, 16).reshape(2, 4)
      tensors = [t00, t01, t10, t11]
      sharded_on_x = layout.Layout(
          [_MESH_DIM_X, layout.UNSHARDED], mesh=self.mesh
      )
      self.assertAllClose(
          np.arange(16).reshape(4, 4),
          numpy_util.unpacked_to_numpy(tensors, sharded_on_x),
      )
  
file_path: tensorflow-upstream/tensorflow/dtensor/python/tests/numpy_util_test.py
imports: import numpy
  from tensorflow.dtensor.python import layout
  from tensorflow.dtensor.python import numpy_util
  from tensorflow.dtensor.python.tests import test_util
  from tensorflow.python.platform import test
label_desc: Use image node_id 3 for calling the NumpyUtilTest obj's underlying member method code with example usage: obj.test_tensor_x_sharded() without return types