class_docstr: Tests for DeformableDETRBackbone.
class_docstr_tok: ["Tests","for","DeformableDETRBackbone","."]
docstr: Test DeformableDETREncoderLayer output shape.
docstr_tok: ["Test","DeformableDETREncoderLayer","output","shape","."]
code: def test_mask_for_shape(self, shape, pad_mask_shape):
      """Test DeformableDETREncoderLayer output shape."""
      pad_mask = None
      if pad_mask_shape is not None:
          pad_mask = jnp.ones(pad_mask_shape, dtype=jnp.bool_)
      x = mask_for_shape(shape, pad_mask=pad_mask)
      self.assertSequenceEqual(x.shape, shape[:3])
  
file_path: scenic/scenic/projects/baselines/deformable_detr/tests/test_backbone.py
imports: import math
  from absl.testing import absltest
  from absl.testing import parameterized
  from jax import random
  import jax.numpy
  from scenic.projects.baselines.deformable_detr.backbone import DeformableDETRBackbone
  from scenic.projects.baselines.deformable_detr.backbone import mask_for_shape
label_desc: Use image node_id 2 for calling the DeformableDETRBackboneTest obj's underlying member method code with example usage: obj.test_mask_for_shape(shape, pad_mask_shape) without return types