(self, x, **kwargs)
| 203 | self.fn = fn |
| 204 | |
| 205 | def forward(self, x, **kwargs): |
| 206 | x = rearrange(x, "b c f ... -> b ... f c") |
| 207 | x, ps = pack_one(x, "* n c") |
| 208 | |
| 209 | o = self.fn(x, **kwargs) |
| 210 | |
| 211 | o = unpack_one(o, ps, "* n c") |
| 212 | return rearrange(o, "b ... f c -> b c f ...") |
| 213 | |
| 214 | |
| 215 | class SqueezeExcite(Module): |
nothing calls this directly
no test coverage detected