(self, x)
| 9 | self.register_buffer('phases', 2 * np.pi * torch.rand(num_channels)) |
| 10 | |
| 11 | def forward(self, x): |
| 12 | y = x.to(torch.float32) |
| 13 | y = y.ger(self.freqs.to(torch.float32)) |
| 14 | y = y + self.phases.to(torch.float32) |
| 15 | y = y.cos() * np.sqrt(2) |
| 16 | return y.to(x.dtype) |
nothing calls this directly
no outgoing calls
no test coverage detected