MCPcopy Create free account
hub / github.com/OpenDriveLab/ReSim / forward

Method forward

sat/vae_modules/autoencoder.py:641–651  ·  view source on GitHub ↗
(
        self,
        x: torch.Tensor,
        input_cp: bool = False,
        latent_cp: bool = False,
        output_cp: bool = False,
        **additional_decode_kwargs,
    )

Source from the content-addressed store, hash-verified

639 return x
640
641 def forward(
642 self,
643 x: torch.Tensor,
644 input_cp: bool = False,
645 latent_cp: bool = False,
646 output_cp: bool = False,
647 **additional_decode_kwargs,
648 ) -> Tuple[torch.Tensor, torch.Tensor, dict]:
649 z, reg_log = self.encode(x, return_reg_log=True, input_cp=input_cp, output_cp=latent_cp)
650 dec = self.decode(z, input_cp=latent_cp, output_cp=output_cp, **additional_decode_kwargs)
651 return z, dec, reg_log

Callers

nothing calls this directly

Calls 2

encodeMethod · 0.95
decodeMethod · 0.95

Tested by

no test coverage detected