docstr: Vectorize module with a single argument.
docstr_tok: ["Vectorize","module","with","a","single","argument","."]
code: def test_vectorize_module_one_arg(self):
      """Vectorize module with a single argument."""
      f = objax.nn.Linear(3, 4)
      fv = objax.Vectorize(f)
      x = objax.random.normal((96, 3))
      y = f(x)
      yv = fv(x)
      self.assertTrue(jn.array_equal(y, yv))
  
file_path: objax/tests/vectorize.py
imports: import unittest
  import jax.numpy
  import numpy
  import objax
label_desc: Use image node_id 1 for calling the TestVectorize obj's underlying member method code with example usage: obj.test_vectorize_module_one_arg() without return types