code: def __next__(self):
      data = next(self.iterator)
      text, length = data.text
      max_length = text.size(1)
      label = data.label - 1
      bs = label.size(0)
      mask = (
          torch.arange(max_length, device=length.device)
          .unsqueeze(0)
          .repeat(bs, 1)
      )
      mask = mask < length.unsqueeze(-1).repeat(1, max_length)
      return (text, mask), label
  
file_path: nni/examples/nas/legacy/textnas/utils.py
imports: import logging
  import torch
  import torch.nn
label_desc: Use image node_id 4 for calling the IteratorWrapper obj's underlying member method code with example usage: obj.__next__() and returns: text,  mask,  label