docstr: Vectorize module with a broadcast argument and a static one.
docstr_tok: ["Vectorize","module","with","a","broadcast","argument","and","a","static","one","."]
code: def test_vectorize_module_one_arg_one_static(self):
      """Vectorize module with a broadcast argument and a static one."""
      f = objax.nn.Linear(3, 4)
      c = objax.random.normal([4])
      fv = objax.Vectorize(
          lambda x, y: f(x) + y, f.vars(), batch_axis=(0, None)
      )
      x = objax.random.normal((96, 3))
      y = f(x) + c
      yv = fv(x, c)
      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 5 for calling the TestVectorize obj's underlying member method code with example usage: obj.test_vectorize_module_one_arg_one_static() without return types