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