code: def test_multi_returns():
      """Test that batching works for a simple circuit with multiple returns"""
      dev = qml.device("default.qubit", wires=2)
  
      @partial(qml.batch_input, argnum=1)
      @qml.qnode(dev, diff_method="parameter-shift")
      def circuit(inputs, weights):
          qml.RY(weights[0], wires=0)
          qml.AngleEmbedding(inputs, wires=range(2), rotation="Y")
          qml.RY(weights[1], wires=1)
          return qml.expval(qml.PauliZ(1)), qml.probs(wires=[0, 1])
  
      batch_size = 6
      inputs = np.random.uniform(
          0, np.pi, (batch_size, 2), requires_grad=False
      )
      weights = np.random.uniform(-np.pi, np.pi, (2,))
  
      res = circuit(inputs, weights)
      assert isinstance(res, tuple)
      assert len(res) == 2
  
      assert res[0].shape == (batch_size,)
      assert res[1].shape == (batch_size, 4)
  
imports: from functools import partial
  import pytest
  import pennylane
  from pennylane import numpy
label_desc: Use image node_id 10 for calling a global function with example usage: test_multi_returns() and returns: qml,  qml