code: def get_split_list(in_dim, child_num):
      in_dim_list = [in_dim // child_num] * child_num
      for _i in range(in_dim % child_num):
          in_dim_list[_i] += 1
      return in_dim_list
  
imports: import os
  import numpy
  import torch.utils.data
  import torchvision.transforms
  import torchvision.datasets
label_desc: Use image node_id 1 for calling a global function with example usage: get_split_list(in_dim, child_num) and returns: in_dim_list