code: def test_flip_sequence_time_major(self):
      x = jnp.arange(2 * 5).reshape((5, 2))
      seq_lengths = jnp.array([4, 2])
  
      flipped = flip_sequences(
          x, seq_lengths, num_batch_dims=1, time_major=True
      )
  
      self.assertEqual(flipped.shape, (5, 2))
      np.testing.assert_allclose(flipped[:4, 0], x[:4, 0][::-1])
      np.testing.assert_allclose(flipped[:2, 1], x[:2, 1][::-1])
  
imports: import jax
  import jax.numpy
  import numpy
  from absl.testing import absltest
  from flax import linen
  from flax.linen.recurrent import flip_sequences
label_desc: Use image node_id 15 for calling the RNNTest obj's underlying member method code with example usage: obj.test_flip_sequence_time_major() without return types