docstr: Test 2d transposed convolution with circular padding on a 3x3 example.
docstr_tok: ["Test","2d","transposed","convolution","with","circular","padding","on","a","3x3","example","."]
code: def test_circular_conv_transpose_2d_custom(self):
      """Test 2d transposed convolution with circular padding on a 3x3 example."""
      rng = dict(params=random.key(0))
      x = np.array(
          (
              (1, 2, 3),
              (4, 5, 6),
              (7, 8, 9),
          )
      )
      x = np.expand_dims(x, (0, 3))
      kernel = np.array(((0, 1, 0), (1, 2, 1), (0, 1, 0)))
      kernel = np.expand_dims(kernel, (2, 3))
  
      conv_module = nn.ConvTranspose(
          features=1,
          kernel_size=(3, 3),
          padding="CIRCULAR",
          kernel_init=lambda *_: kernel,
          bias_init=initializers.zeros,
      )
      y, initial_params = conv_module.init_with_output(rng, x)
  
      self.assertEqual(
          initial_params["params"]["kernel"].shape, (3, 3, 1, 1)
      )
      # Compare with manually computed convolution
      correct_ans = np.array(
          (
              (18, 21, 24),
              (27, 30, 33),
              (36, 39, 42),
          )
      )
      correct_ans = np.expand_dims(correct_ans, (0, 3))
      np.testing.assert_allclose(y, correct_ans)
  
imports: import functools
  import jax
  import jax.numpy
  import numpy
  from absl.testing import absltest, parameterized
  from jax import random
  from jax.nn import initializers
  from flax import linen
label_desc: Use image node_id 37 for calling the LinearTest obj's underlying member method code with example usage: obj.test_circular_conv_transpose_2d_custom() without return types