(
self, x: torch.Tensor, encoded: Optional[Dict[str, torch.Tensor]]
)
| 129 | return dict(latents=h) |
| 130 | |
| 131 | def predict_sdf( |
| 132 | self, x: torch.Tensor, encoded: Optional[Dict[str, torch.Tensor]] |
| 133 | ) -> torch.Tensor: |
| 134 | data = encoded["latents"] |
| 135 | x = self.decoder_input_proj(x.permute(0, 2, 1)) |
| 136 | x = self.decoder(x, data) |
| 137 | x = self.ln_post(x) |
| 138 | x = self.output_proj(x) |
| 139 | return x[..., 0] |
nothing calls this directly
no outgoing calls
no test coverage detected