code: def emformer_data_gen_fn():
      input = torch.rand(4, 400, IN_FEATURES)
      lengths = torch.randint(1, 200, (4,))
      return dict(input=input, lengths=lengths)
  
file_path: ColossalAI/tests/kit/model_zoo/torchaudio/torchaudio.py
imports: from functools import partial
  import torch
  import torchaudio.models
  from ..registry import ModelAttribute, model_zoo
label_desc: Use image node_id 2 for calling a global function with example usage: emformer_data_gen_fn() and returns: dict