code: def test_channels_first_has_correct_output(self, rank):
      # This tests that the layer produces the correct output for a number of
      # different input dimensionalities with 'channels_first' data format.
      x = tf.random.uniform((6, 5, 4, 3, 2, 1)[:rank], dtype=tf.float32)
      y = gdn.GDN(
          inverse=False, rectify=False, data_format="channels_first"
      )(x)
      self.assertEqual(x.shape, y.shape)
      self.assertAllClose(y, x / (1 + 0.1 * abs(x)), rtol=0, atol=1e-6)
  
imports: import os
  from absl.testing import parameterized
  import tensorflow
  from tensorflow_compression.python.layers import gdn
  from tensorflow_compression.python.layers import parameters
label_desc: Use image node_id 5 for calling the GDNTest obj's underlying member method code with example usage: obj.test_channels_first_has_correct_output(rank) without return types