class_docstr: Class to test the shape of the Grad/Jacobian/Hessian with different return types and shot vectors.
class_docstr_tok: ["Class","to","test","the","shape","of","the","Grad\/Jacobian\/Hessian","with","different","return","types","and","shot","vectors","."]
docstr: The jacobian of multiple measurements with a multiple params array return a single array.
docstr_tok: ["The","jacobian","of","multiple","measurements","with","a","multiple","params","array","return","a","single","array","."]
code: def test_jacobian_multiple_measurement_multiple_param_array(
      self,
      dev_name,
      diff_method,
      gradient_kwargs,
      shots,
      jacobian,
      interface,
  ):
      """The jacobian of multiple measurements with a multiple params array return a single array."""
      dev = qml.device(dev_name, wires=2, shots=shots)
  
      @qnode(
          dev,
          interface=interface,
          diff_method=diff_method,
          **gradient_kwargs,
      )
      def circuit(a):
          qml.RY(a[0], wires=0)
          qml.RX(a[1], wires=0)
          return qml.expval(qml.PauliZ(0)), qml.probs(wires=[0, 1])
  
      a = jax.numpy.array([0.1, 0.2])
  
      jac = jacobian(circuit)(a)
  
      num_copies = sum(
          [1 for x in shots if isinstance(x, int)]
          + [x[1] for x in shots if isinstance(x, tuple)]
      )
      assert len(jac) == num_copies
      for j in jac:
          assert isinstance(j, tuple)
          assert len(j) == 2  # measurements
  
          assert isinstance(j[0], jax.numpy.ndarray)
          assert j[0].shape == (2,)
  
          assert isinstance(j[1], jax.numpy.ndarray)
          assert j[1].shape == (4, 2)
  
imports: import pytest
  from flaky import flaky
  import pennylane
  from pennylane import numpy
  from pennylane import qnode
label_desc: Use image node_id 13 for calling the TestReturnWithShotVectors obj's underlying member method code with example usage: obj.test_jacobian_multiple_measurement_multiple_param_array(dev_name, diff_method, gradient_kwargs, shots, jacobian, interface) and returns: qml,  qml