code: def rosenbrock(tensor):
      # https://en.wikipedia.org/wiki/Test_functions_for_optimization
      x, y = tensor
      return (1 - x) ** 2 + 100 * (y - x**2) ** 2
  
file_path: pytorch-optimizer/examples/viz_optimizers.py
imports: import math
  import matplotlib.pyplot
  import numpy
  import torch
  from hyperopt import fmin, hp, tpe
  import torch_optimizer
label_desc: Use image node_id 1 for calling a global function with example usage: rosenbrock(tensor) and returns: unknown