code: def check_mkdir(dir_name):
      if not os.path.exists(dir_name):
          os.mkdir(dir_name)
  
imports: import os
  from math import ceil
  import numpy
  import torch
  import torch.nn.functional
  from torch import nn
label_desc: Use image node_id 1 for calling a global function with example usage: check_mkdir(dir_name) without return types