Method
__init__
(self, x_dim, min_deg, max_deg, use_identity: bool = True)
Source from the content-addressed store, hash-verified
| 8 | """Sinusoidal Positional Encoder used in Nerf.""" |
| 9 | |
| 10 | def __init__(self, x_dim, min_deg, max_deg, use_identity: bool = True): |
| 11 | super().__init__() |
| 12 | self.x_dim = x_dim |
| 13 | self.min_deg = min_deg |
| 14 | self.max_deg = max_deg |
| 15 | self.use_identity = use_identity |
| 16 | self.register_buffer( |
| 17 | "scales", torch.tensor([2**i for i in range(min_deg, max_deg)]) |
| 18 | ) |
| 19 | |
| 20 | @property |
| 21 | def latent_dim(self) -> int: |
Callers
nothing calls this directly
Tested by
no test coverage detected