(self, x)
| 358 | return x |
| 359 | |
| 360 | def wavelet_transform(self, x): |
| 361 | xa = torch.cat([x[:, ::2, :, :], |
| 362 | x[:, 1::2, :, :], |
| 363 | ], -1) |
| 364 | d = torch.matmul(xa, self.ec_d) |
| 365 | s = torch.matmul(xa, self.ec_s) |
| 366 | return d, s |
| 367 | |
| 368 | def evenOdd(self, x): |
| 369 |