def encode_to_channels(
        self, batch: AttrDict, options: Optional[AttrDict] = None
    ) -> torch.Tensor:
        """
        Encode the batch into a per-data-point set of latents.
        :return: [batch_size, latent_ctx, latent_width]
        """

        Encode the batch into a per-data-point set of latents.
        :return: [batch_size, latent_ctx, latent_width]
        
shap_e/models/transmitter/base.py