code: def make_crypten_compatible(initialization_function):
      def wrapper_func(tensor, *args, **kwargs):
          if not torch.is_tensor(tensor):
              result = torch.empty(tensor.size())
              result = initialization_function(result, *args, **kwargs)
              tensor.set(result)
              return tensor
  
          return initialization_function(tensor, *args, **kwargs)
  
      return wrapper_func
  
label_desc: Use image node_id 1 for calling a global function with example usage: make_crypten_compatible(initialization_function) and returns: wrapper_func, initialization_function, tensor