code: def test_sharded_embedding_bag_colwise(self):
      for spec in generate_chunk_sharding_specs_for_test(1):
          self._test_sharded_embedding_bag_with_test_cases(spec, 1)
  
file_path: pytorch/test/distributed/_shard/sharded_tensor/ops/test_embedding_bag.py
imports: import sys
  import torch
  import torch.distributed
  from torch.distributed._shard import shard_parameter
  from torch.testing._internal.common_distributed import requires_nccl, skip_if_lt_x_gpu
  from torch.testing._internal.common_utils import TEST_WITH_DEV_DBG_ASAN, run_tests
  from torch.testing._internal.distributed._shard.sharded_tensor import TEST_GPU_NUM, ShardedTensorTestBase, with_comms
  from torch.testing._internal.distributed._shard.sharded_tensor._test_ops_common import clone_module_parameter, generate_chunk_sharding_specs_for_test, generate_local_weight_sharding_params_for_test
label_desc: Use image node_id 2 for calling the TestShardedEmbeddingBag obj's underlying member method code with example usage: obj.test_sharded_embedding_bag_colwise() without return types