code: def gray2d_setup():
      images = {}
  
      x = th.zeros(1, 30, 30)
      x[:, 10:21, 10:21] = 1
      images["gray_01"] = x
  
      x = th.zeros(1, 30, 40)
      x[:, 10:21, 10:21] = 1
      images["gray_02"] = x
      return images
  
imports: import torch
  from pywick.transforms import ToTensor, ToFile, ChannelsLast, HWC, ChannelsFirst, CHW, TypeCast, AddChannel, Transpose, RangeNormalize, StdNormalize, RandomCrop, SpecialCrop, Pad, RandomFlip, RandomOrder
label_desc: Use image node_id 1 for calling a global function with example usage: gray2d_setup() and returns: images