code: def create_input(shape, offset: int = 0):
      x = np.arange(np.prod(shape)).reshape(shape) + offset
  
      return torch.from_numpy(x).to(torch.get_default_dtype())
  
imports: import pickle
  import numpy
  import tensorflow
  import tensorflow_addons
  import torch
label_desc: Use image node_id 1 for calling a global function with example usage: create_input(shape, offset) and returns: torch