class_docstr: Tests for ViT+ utilities.
class_docstr_tok: ["Tests","for","ViT+","utilities","."]
code: def test_compute_box_bias_size(self):
      feature_map = jnp.ones((2, 3, 3, 10))
      box_bias = utils.compute_box_bias(feature_map, kind="size")
      expected_bias = jnp.array(
          [
              [0.0, 0.0, -0.692997, -0.692997],
              [0.0, 0.0, -0.692997, -0.692997],
              [0.0, 0.0, -0.692997, -0.692997],
              [0.0, 0.0, -0.692997, -0.692997],
              [0.0, 0.0, -0.692997, -0.692997],
              [0.0, 0.0, -0.692997, -0.692997],
              [0.0, 0.0, -0.692997, -0.692997],
              [0.0, 0.0, -0.692997, -0.692997],
              [0.0, 0.0, -0.692997, -0.692997],
          ],
      )
      np.testing.assert_allclose(expected_bias, box_bias, 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 5 for calling the UtilsTest obj's underlying member method code with example usage: obj.test_compute_box_bias_size() without return types