code: def rosenbrock(tensor):
      x, y = tensor
      return (1 - x) ** 2 + 100 * (y - x**2) ** 2
  
imports: import math
  import pytest
  import functools
  from copy import deepcopy
  import torch
  from torch.testing._internal.common_utils import TestCase
  from torch.nn import Parameter
  from timm.scheduler import PlateauLRScheduler
  from timm.optim import create_optimizer_v2
  import importlib
  import os
label_desc: Use image node_id 5 for calling a global function with example usage: rosenbrock(tensor) and returns: unknown