code: def rosenbrock(tensor):
      x, y = tensor
      return (1 - x) ** 2 + 1 * (y - x**2) ** 2
  
file_path: pytorch-optimizer/tests/test_basic.py
imports: import pytest
  import torch
  import torch_optimizer
label_desc: Use image node_id 1 for calling a global function with example usage: rosenbrock(tensor) and returns: unknown