docstr: Pass an input through a linear filter with 3 units and bias
  test the shape and contents of the output.
docstr_tok: ["Pass","an","input","through","a","linear","filter","with","3","units","and","bias","test","the","shape","and","contents","of","the","output","."]
code: def test_on_linear_three_unit_with_bias(self):
      """
      Pass an input through a linear filter with 3 units and bias
      test the shape and contents of the output.
      """
  
      # Define linear filter with 1 input channel and 3 output channels
      linear_filter = objax.nn.Linear(1, 3, use_bias=True)
      weights = objax.TrainVar(jn.array([[1.0, 2.0, 1.0]]))
      bias = objax.TrainVar(jn.array([2.0, 1.0, 2.0]))
      linear_filter.w = weights
      linear_filter.b = bias
  
      # Define data and compute output response of linear filter
      data = jn.array([[1.0], [2.0]])
      features = linear_filter(data)
      expected_features = jn.array([[3.0, 3.0, 3.0], [4.0, 5.0, 4.0]])
      self.assertEqual(features.shape, (2, 3))
      self.assertTrue(jn.array_equal(features, expected_features))
  
imports: import unittest
  import jax.numpy
  import objax
label_desc: Use image node_id 2 for calling the TestLinear obj's underlying member method code with example usage: obj.test_on_linear_three_unit_with_bias() without return types