code: def test_sharded_embedding_colwise(self):
      self._run_embedding_op_test(1, [5, 4], 17, 12)
      self._run_embedding_op_test(1, [6, 7, 6], 21, 11)
      self._run_embedding_op_test(1, [8, 6, 5, 4], 23, 13)
      self._run_embedding_op_test(1, [8, 6, 5, 4, 7], 23, 16)
      self._run_embedding_op_test(1, [4], 15, 14)
      self._run_embedding_op_test(1, [34], 15, 14, padding_idx=10)
      self._run_embedding_op_test(
          1, [8, 6, 5, 4], 23, 13, padding_idx=12
      )
  
file_path: pytorch/test/distributed/_tensor/test_embedding_ops.py
imports: import sys
  import torch
  from torch.distributed._tensor import DTensor
  from torch.distributed._tensor.placement_types import Replicate
  from torch.distributed.tensor.parallel import ColwiseParallel, parallelize_module, RowwiseParallel
  from torch.testing._internal.common_utils import run_tests, TEST_WITH_DEV_DBG_ASAN
  from torch.testing._internal.distributed._tensor.common_dtensor import DTensorTestBase, with_comms
label_desc: Use image node_id 2 for calling the TestEmbeddingOp obj's underlying member method code with example usage: obj.test_sharded_embedding_colwise() without return types