code: def test_mtv_keep_spatiotemporal_features(
      self, final_endpoint, expected_output_shape
  ):
      rngs = {"params": random.PRNGKey(0), "dropout": random.PRNGKey(1)}
      images = jnp.ones((2, 4, 8, 8, 3))
      outputs, _ = model.MTV(
          view_configs=self.view_configs,
          cross_view_fusion=self.cross_view_fusion,
          temporal_encoding_config=self.temporal_encoding_config,
          global_encoder_config=self.global_encoder_config,
          input_token_temporal_dims=[1, 2],
          num_classes=10,
          classifier="gap",
          keep_spatiotemporal_features=True,
          final_endpoint=final_endpoint,
      ).init_with_output(rngs, images)
      self.assertEqual(outputs.shape, expected_output_shape)
  
imports: from absl.testing import parameterized
  import flax
  from jax import random
  import jax.numpy
  import ml_collections
  from scenic.projects.mtv import model
  import tensorflow
label_desc: Use image node_id 8 for calling the ModelTest obj's underlying member method code with example usage: obj.test_mtv_keep_spatiotemporal_features(final_endpoint, expected_output_shape) without return types