code: def get_mask(
      pad_size: torch.Size, lengths: torch.Tensor
  ) -> torch.Tensor:
      device = lengths.device
      return torch.arange(pad_size, device=device)[None, :] < lengths[
          :, None
      ].to(device)
  
label_desc: Use image node_id 2 for calling a global function with example usage: get_mask(pad_size, lengths) and returns: unknown