code: def test_forward_shape(self):
      layer = tl.AvgPool(pool_size=(2, 2), strides=(1, 2))
      x = np.ones((11, 6, 4, 17))
      y = layer(x)
      self.assertEqual(y.shape, (11, 5, 2, 17))
  
file_path: trax/trax/layers/pooling_test.py
imports: from absl.testing import absltest
  import numpy
  import trax.layers
label_desc: Use image node_id 1 for calling the AvgPoolTest obj's underlying member method code with example usage: obj.test_forward_shape() without return types