class_docstr: Tests for ViT+ utilities.
class_docstr_tok: ["Tests","for","ViT+","utilities","."]
code: def test_seq2img(self, img_shape, features_shape, expected_shape):
      images = jnp.ones(img_shape)
      features = jnp.ones(features_shape)
      features_2d = utils.seq2img(images, features)
      self.assertEqual(expected_shape, features_2d.shape)
  
file_path: scenic/scenic/projects/owl_vit/tests/utils_test.py
imports: from absl.testing import absltest
  from absl.testing import parameterized
  import jax
  import jax.numpy
  import numpy
  from scenic.projects.owl_vit import utils
label_desc: Use image node_id 1 for calling the UtilsTest obj's underlying member method code with example usage: obj.test_seq2img(img_shape, features_shape, expected_shape) without return types