code: def test_reverse_shuffle_layer(self):
      reverse_shuffle_layer = rse.ReverseShuffleLayer()
      x = np.array([[[0], [1], [2], [3], [4], [5], [6], [7]]])
      print(x.shape)
      _, _ = reverse_shuffle_layer.init(shapes.signature(x))
      y = reverse_shuffle_layer(x)
      expected_output = np.array(
          [[[0], [4], [1], [5], [2], [6], [3], [7]]]
      )
      self._assert_equal_tensors(y, expected_output)
  
file_path: trax/trax/models/research/rse_test.py
imports: from absl.testing import absltest
  import numpy
  from trax import shapes
  from trax.models.research import rse
label_desc: Use image node_id 4 for calling the RSETest obj's underlying member method code with example usage: obj.test_reverse_shuffle_layer() without return types