(self, x)
| 53 | nn.init.constant_(m.bias, 0) |
| 54 | |
| 55 | def forward(self, x): |
| 56 | res_list = [] |
| 57 | for i in range(self.num_kernels + 1): |
| 58 | res_list.append(self.kernels[i](x)) |
| 59 | res = torch.stack(res_list, dim=-1).mean(-1) |
| 60 | return res |
nothing calls this directly
no outgoing calls
no test coverage detected