code: def get_basic_data(device):
      # Example provided
      # in 6f73a2513dc784c59eec153a45f40bc528355b18
      # of https://github.com/HawkAaron/warp-transducer
  
      logits = torch.tensor(
          [
              [
                  [
                      [0.1, 0.6, 0.1, 0.1, 0.1],
                      [0.1, 0.1, 0.6, 0.1, 0.1],
                      [0.1, 0.1, 0.2, 0.8, 0.1],
                  ],
                  [
                      [0.1, 0.6, 0.1, 0.1, 0.1],
                      [0.1, 0.1, 0.2, 0.1, 0.1],
                      [0.7, 0.1, 0.2, 0.1, 0.1],
                  ],
              ]
          ],
          dtype=torch.float32,
          device=device,
      )
      targets = torch.tensor([[1, 2]], dtype=torch.int, device=device)
      logit_lengths = torch.tensor([2], dtype=torch.int, device=device)
      target_lengths = torch.tensor([2], dtype=torch.int, device=device)
  
      logits.requires_grad_(True)
  
      return logits, targets, logit_lengths, target_lengths
  
imports: import random
  import unittest
  import numpy
  import torch
  from torchaudio.functional import rnnt_loss
label_desc: Use image node_id 3 for calling a global function with example usage: get_basic_data(device) and returns: logits,  targets,  logit_lengths,  target_lengths