class_docstr: Class to test the shape of the Grad/Jacobian/Hessian with different return types.
class_docstr_tok: ["Class","to","test","the","shape","of","the","Grad\/Jacobian\/Hessian","with","different","return","types","."]
docstr: For a multi dimensional measurement (probs), check that a single tuple is returned containing arrays with
  the correct dimension
docstr_tok: ["For","a","multi","dimensional","measurement","(","probs",")",",","check","that","a","single","tuple","is","returned","containing","arrays","with","the","correct","dimension"]
code: def test_jacobian_single_measurement_probs_multiple_param(
      self, dev, diff_method, grad_on_execution, device_vjp, shots
  ):
      """For a multi dimensional measurement (probs), check that a single tuple is returned containing arrays with
      the correct dimension"""
      if diff_method == "adjoint":
          pytest.skip(
              "Test does not supports adjoint because of probabilities."
          )
      if shots is not None and diff_method in ("backprop", "adjoint"):
          pytest.skip(
              "Test does not support finite shots and adjoint/backprop"
          )
  
      @qnode(
          dev,
          interface="torch",
          diff_method=diff_method,
          device_vjp=device_vjp,
          grad_on_execution=grad_on_execution,
      )
      def circuit(a, b):
          qml.RY(a, wires=0)
          qml.RX(b, wires=0)
          return qml.probs(wires=[0, 1])
  
      a = torch.tensor(0.1, requires_grad=True)
      b = torch.tensor(0.2, requires_grad=True)
  
      jac = jacobian(
          lambda _a, _b: circuit(_a, _b, shots=shots), (a, b)
      )
  
      assert isinstance(jac, tuple)
  
      assert isinstance(jac[0], torch.Tensor)
      assert jac[0].shape == (4,)
  
      assert isinstance(jac[1], torch.Tensor)
      assert jac[1].shape == (4,)
  
imports: import numpy
  import pytest
  import pennylane
  from pennylane import qnode
  from pennylane.devices import DefaultQubit
label_desc: Use image node_id 5 for calling the TestReturn obj's underlying member method code with example usage: obj.test_jacobian_single_measurement_probs_multiple_param(dev, diff_method, grad_on_execution, device_vjp, shots) and returns: qml