class_docstr: Tests for the jacobian function
docstr: Test the qml.jacobian function for multiple argnums
docstr_tok: ["Test","the","qml.jacobian","function","for","multiple","argnums"]
code: def test_multiple_argnum_jacobian(self, tol):
      """Test the qml.jacobian function for multiple argnums"""
      cost_fn = lambda x, y: np.array(
          [np.sin(x) * np.cos(y), x * y**2]
      )
  
      x = np.array(0.5, requires_grad=True)
      y = np.array(0.2, requires_grad=True)
  
      jac_fn = qml.jacobian(cost_fn, argnum=[0, 1])
      res = jac_fn(x, y)
      expected = (
          np.array([np.cos(x) * np.cos(y), y**2]),
          np.array([-np.sin(x) * np.sin(y), 2 * x * y]),
      )
      assert all(
          np.allclose(_r, _e, atol=tol, rtol=0)
          for _r, _e in zip(res, expected)
      )
  
imports: import autograd
  import pytest
  import pennylane
  from pennylane import numpy
label_desc: Use image node_id 2 for calling the TestJacobian obj's underlying member method code with example usage: obj.test_multiple_argnum_jacobian(tol) without return types