(self, num_channels, bandwidth=1)
| 4 | |
| 5 | class MPFourier(torch.nn.Module): |
| 6 | def __init__(self, num_channels, bandwidth=1): |
| 7 | super().__init__() |
| 8 | self.register_buffer('freqs', 2 * np.pi * torch.randn(num_channels) * bandwidth) |
| 9 | self.register_buffer('phases', 2 * np.pi * torch.rand(num_channels)) |
| 10 | |
| 11 | def forward(self, x): |
| 12 | y = x.to(torch.float32) |
nothing calls this directly
no outgoing calls
no test coverage detected