(self, input, **kwargs)
| 27 | self.freq_bands = self.freq_bands.numpy().tolist() |
| 28 | |
| 29 | def forward(self, input, **kwargs): |
| 30 | |
| 31 | out = [] |
| 32 | if self.include_input: |
| 33 | out.append(input) |
| 34 | |
| 35 | for i in range(len(self.freq_bands)): |
| 36 | freq = self.freq_bands[i] |
| 37 | for p_fn in self.periodic_fns: |
| 38 | out.append(p_fn(input * freq)) |
| 39 | |
| 40 | out = torch.cat(out, dim=-1) |
| 41 | |
| 42 | |
| 43 | return out |
| 44 | |
| 45 | def get_encoder(encoding, input_dim=3, |
| 46 | multires=6, |
nothing calls this directly
no outgoing calls
no test coverage detected