code: def Q_batch(theta):
      """Returns the true function value for a batch of parameters with size (B, 2)"""
      return 1.2 - (3 / 4 * theta[:, 0] ** 2 + theta[:, 1] ** 2)
  
file_path: ray/doc/source/tune/examples/pbt_visualization/pbt_visualization_utils.py
imports: import matplotlib.pyplot
  from matplotlib.animation import FuncAnimation, PillowWriter
  import numpy
label_desc: Use image node_id 2 for calling a global function with example usage: Q_batch(theta) and returns: unknown