code: def get_ebc():
      # EmbeddingBagCollection
      eb1_config = EmbeddingBagConfig(
          name="t1",
          embedding_dim=SHAPE,
          num_embeddings=SHAPE,
          feature_names=["f1"],
      )
      eb2_config = EmbeddingBagConfig(
          name="t2",
          embedding_dim=SHAPE,
          num_embeddings=SHAPE,
          feature_names=["f2"],
      )
      return EmbeddingBagCollection(
          tables=[eb1_config, eb2_config], device=torch.device("cpu")
      )
  
file_path: colossalai/tests/kit/model_zoo/torchrec/torchrec.py
imports: from functools import partial
  import torch
  from torchrec.models import deepfm, dlrm
  from torchrec.modules.embedding_configs import EmbeddingBagConfig
  from torchrec.modules.embedding_modules import EmbeddingBagCollection
  from torchrec.sparse.jagged_tensor import KeyedJaggedTensor, KeyedTensor
  from ..registry import model_zoo
label_desc: Use image node_id 7 for calling a global function with example usage: get_ebc() and returns: EmbeddingBagCollection