class_docstr: Tests for ViT+ utilities.
class_docstr_tok: ["Tests","for","ViT+","utilities","."]
code: def test_normalized_grid_corner_coordinates(self):
      feature_map = jnp.ones((2, 3, 3, 10))
      corner_coords = utils.normalized_grid_corner_coordinates(
          feature_map, None
      )
      expected_coords = jnp.array(
          [
              [0.33333334, 0.33333334],
              [0.6666667, 0.33333334],
              [1.0, 0.33333334],
              [0.33333334, 0.6666667],
              [0.6666667, 0.6666667],
              [1.0, 0.6666667],
              [0.33333334, 1.0],
              [0.6666667, 1.0],
              [1.0, 1.0],
          ]
      )
      np.testing.assert_allclose(
          expected_coords, corner_coords, atol=1e-5
      )
  
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 2 for calling the UtilsTest obj's underlying member method code with example usage: obj.test_normalized_grid_corner_coordinates() without return types