code: def tacotron_data_gen_fn():
      n_batch = 4
      max_text_length = 100
      max_mel_specgram_length = 300
      tokens = torch.randint(0, 148, (n_batch, max_text_length))
      token_lengths = max_text_length * torch.ones((n_batch,))
      mel_specgram = torch.rand(
          n_batch, N_MELS, max_mel_specgram_length
      )
      mel_specgram_lengths = max_mel_specgram_length * torch.ones(
          (n_batch,)
      )
      return dict(
          tokens=tokens,
          token_lengths=token_lengths,
          mel_specgram=mel_specgram,
          mel_specgram_lengths=mel_specgram_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 4 for calling a global function with example usage: tacotron_data_gen_fn() and returns: dict