Method encode_to_vector

shap_e/models/transmitter/base.py:70–73

Encode the batch into a single latent vector.

(self, batch: AttrDict, options: Optional[AttrDict] = None)

    @abstractmethod
    def encode_to_vector(self, batch: AttrDict, options: Optional[AttrDict] = None) -> torch.Tensor:
        """
        Encode the batch into a single latent vector.
        """

    def bottleneck_to_params(
        self, vector: torch.Tensor, options: Optional[AttrDict] = None