Method encode_to_vector

shap_e/models/transmitter/base.py:113–114

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

        """

    def encode_to_vector(self, batch: AttrDict, options: Optional[AttrDict] = None) -> torch.Tensor:
        return self.encode_to_channels(batch, options=options).flatten(1)

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