code: def test_trainvar_assign_multivalue(self):
      m = objax.ModuleList([objax.TrainVar(jn.array((1.0, 2.0)))])
  
      def increase(x):
          m[0].assign(m[0].value + x)
          return x * 2
  
      x = np.arange(10)[:, None]
      vec_increase = objax.Vectorize(increase, m.vars())
      y = vec_increase(x)
      self.assertEqual(
          y.tolist(), (2 * np.arange(10))[:, None].tolist()
      )
      self.assertEqual(m[0].value.tolist(), [5.5, 6.5])
  
file_path: objax/tests/vectorize.py
imports: import unittest
  import jax.numpy
  import numpy
  import objax
label_desc: Use image node_id 13 for calling the TestVectorize obj's underlying member method code with example usage: obj.test_trainvar_assign_multivalue() and returns: unknown