code: def expected_probs(params):
      """Expected result of the below circuit ansatz."""
      x, y = params[..., 0], params[..., 1]
      c_x, s_x = np.cos(x / 2), np.sin(x / 2)
      c_y, s_y = np.cos(y / 2), np.sin(y / 2)
      # Transpose to put potential broadcasting axis first
      return (
          np.array([c_x * c_y, c_x * s_y, s_x * s_y, s_x * c_y]).T ** 2
      )
  
imports: import pytest
  import pennylane
  from pennylane import numpy
  from pennylane.gradients import param_shift
label_desc: Use image node_id 1 for calling a global function with example usage: expected_probs(params) and returns: unknown