code: def _default_labels_split_fn(labels: torch.Tensor):
      if isinstance(labels, torch.Tensor):
          return [
              sl.squeeze(0)
              for sl in torch.split(labels.detach().cpu(), 1)
          ]
      else:
          raise NotImplementedError(
              f"Only support split tensor, please customize split function for {type(labels)}."
          )
  
file_path: nni/nni/contrib/distillation/utils.py
imports: from pathlib import Path
  import pickle
  import random
  from typing import Dict, List
  import numpy
  import torch
label_desc: Use image node_id 2 for calling a global function with example usage: _default_labels_split_fn(labels) and returns: unknown