(self, input)
| 426 | self.register_buffer('weight', torch.randn([out_features // 2, in_features]) * std) |
| 427 | |
| 428 | def forward(self, input): |
| 429 | f = 2 * math.pi * input @ self.weight.T |
| 430 | return torch.cat([f.cos(), f.sin()], dim=-1) |
| 431 | |
| 432 | |
| 433 | # U-Nets |
nothing calls this directly
no outgoing calls
no test coverage detected