code: def interaction_arch_data_gen_fn():
      KT = gen_kt()
      return dict(
          dense_features=torch.rand((BATCH, SHAPE)), sparse_features=KT
      )
  
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 3 for calling a global function with example usage: interaction_arch_data_gen_fn() and returns: dict