docstr: Pass an input through two-by-two convolution filter
  with two channels and test the shape and contents of the output.
docstr_tok: ["Pass","an","input","through","two-by-two","convolution","filter","with","two","channels","and","test","the","shape","and","contents","of","the","output","."]
code: def test_on_conv_transpose_2d_two_channels(self):
      """
      Pass an input through two-by-two convolution filter
      with two channels and test the shape and contents of the output.
      """
  
      # Channels/Colors, #filters, filter_size (square)
      conv_filter = objax.nn.ConvTranspose2D(
          2, 1, 2, padding=objax.ConvPadding.VALID
      )
      weights = objax.TrainVar(
          jn.array(
              [[[[1.0, 2.0], [3.0, 4.0]], [[1.0, 2.0], [3.0, 4.0]]]]
          ).transpose((2, 3, 0, 1))
      )
      conv_filter.w = weights
      image = jn.array(
          [
              [
                  [
                      [2.0, 1.0, 3.0, 4.0],
                      [5.0, 6.0, 7.0, 8.0],
                      [9.0, 10.0, 11.0, 12.0],
                      [13.0, 14.0, 15.0, 16.0],
                  ],
                  [
                      [2.0, 1.0, 3.0, 4.0],
                      [5.0, 6.0, 7.0, 8.0],
                      [9.0, 10.0, 11.0, 12.0],
                      [13.0, 14.0, 15.0, 16.0],
                  ],
              ]
          ]
      )
      # NCHW: Batch, Channels/Colors, Height, Width
      features = conv_filter(image)
      expected_features = jn.array(
          [
              [
                  [
                      [4.0, 10.0, 10.0, 20.0, 16.0],
                      [22.0, 54.0, 64.0, 92.0, 64.0],
                      [48.0, 132.0, 152.0, 172.0, 112.0],
                      [80.0, 212.0, 232.0, 252.0, 160.0],
                      [78.0, 188.0, 202.0, 216.0, 128.0],
                  ]
              ]
          ]
      )
      self.assertEqual(features.shape, (1, 1, 5, 5))
      self.assertTrue(jn.array_equal(features, expected_features))
  
file_path: objax/tests/conv_transpose.py
imports: import unittest
  import jax.numpy
  import objax
label_desc: Use image node_id 6 for calling the TestConvTranspose obj's underlying member method code with example usage: obj.test_on_conv_transpose_2d_two_channels() without return types