docstr: Parallel inference (concat reduction) for multiple outputs.
docstr_tok: ["Parallel","inference","(","concat","reduction",")","for","multiple","outputs","."]
code: def test_parallel_concat_multi_output(self):
      """Parallel inference (concat reduction) for multiple outputs."""
      f = objax.nn.Linear(3, 4)
      x = objax.random.normal((96, 3))
      y = f(x)
      fp = objax.Parallel(lambda x: [f(x), f(-x)], vc=f.vars())
      with fp.vars().replicate():
          z1, z2 = fp(x)
      self.assertTrue(jn.array_equal(z1, y))
      self.assertTrue(jn.array_equal(z2, -y))
  
imports: import os
  import unittest
  import jax.numpy
  import numpy
  import objax
label_desc: Use image node_id 3 for calling the TestParallel obj's underlying member method code with example usage: obj.test_parallel_concat_multi_output() without return types