code: def batched_rosenbrock(
      x: np.ndarray, num_batches: int, a: np.ndarray, b: np.ndarray
  ) -> np.ndarray:
      """A batched version of the rosenbrock example"""
  
      fall = np.zeros(num_batches)
      for i in range(num_batches):
          fall[i] = rosenbrock(x[i * 2 : (i + 1) * 2], a[i], b[i])
  
      return fall
  
imports: from cuml.tsa.batched_lbfgs import batched_fmin_lbfgs_b
  from cuml.internals.safe_imports import cpu_only_import
label_desc: Use image node_id 3 for calling a global function with example usage: batched_rosenbrock(x, num_batches, a, b) and returns: fall