code: def gpu(tensor, gpu=False):
      if gpu:
          return tensor.cuda()
      else:
          return tensor
  
imports: import numpy
  import torch
label_desc: Use image node_id 1 for calling a global function with example usage: gpu(tensor, gpu) and returns: tensor, tensor