class_docstr: A two-tower image-text encoder model.
class_docstr_tok: ["A","two-tower","image-text","encoder","model","."]
code: def __call__(  # pytype: disable=signature-mismatch  # jax-ndarray
      self,
      image: jnp.ndarray,  # Shape: (N, H, W, C)
      text: jnp.ndarray,  # Shape: (N, L)
      mask: Optional[jnp.ndarray] = None,  # Shape: (N, L)
      *,
      train: bool = False,
      debug: bool = False,
  ) -> tuple[jnp.ndarray, jnp.ndarray]:  # Shapes: (N, E) and (N, E)
      return (
          self.image_encoder(
              image, train=train, debug=debug
          ),  # pylint: disable=not-callable
          self.text_encoder(text, mask, train=train, debug=debug),
      )
  
imports: import abc
  from typing import Literal
  from typing import Optional
  import flax
  from flax.core.frozen_dict import FrozenDict
  import flax.linen
  import jax
  import jax.numpy
label_desc: Use image node_id 1 for calling the ImageTextEncoder obj's underlying member method code with example usage: obj.__call__(image, text, mask) and returns: self,  self