(
self,
cfg: DecoderSplattingCUDACfg,
)
| 23 | background_color: Float[Tensor, "3"] |
| 24 | |
| 25 | def __init__( |
| 26 | self, |
| 27 | cfg: DecoderSplattingCUDACfg, |
| 28 | ) -> None: |
| 29 | super().__init__(cfg) |
| 30 | self.make_scale_invariant = cfg.make_scale_invariant |
| 31 | self.register_buffer( |
| 32 | "background_color", |
| 33 | torch.tensor(cfg.background_color, dtype=torch.float32), |
| 34 | persistent=False, |
| 35 | ) |
| 36 | |
| 37 | def forward( |
| 38 | self, |
nothing calls this directly
no outgoing calls
no test coverage detected