(self)
| 311 | self.register_buffer('fourier_coefs', fourier_coefs) # [1, num_fourier_feats] |
| 312 | |
| 313 | def get_dim(self) -> int: |
| 314 | return self.fourier_coefs.shape[1] * 2 |
| 315 | |
| 316 | def forward(self, t: torch.Tensor) -> torch.Tensor: |
| 317 | assert t.ndim == 2, f"Wrong shape: {t.shape}" |
nothing calls this directly
no outgoing calls
no test coverage detected