(self, x)
| 368 | return x |
| 369 | |
| 370 | def wavelet_transform(self, x): |
| 371 | xa = torch.cat([x[:, ::2, :, :], |
| 372 | x[:, 1::2, :, :], |
| 373 | ], -1) |
| 374 | d = torch.matmul(xa, self.ec_d) |
| 375 | s = torch.matmul(xa, self.ec_s) |
| 376 | return d, s |
| 377 | |
| 378 | def evenOdd(self, x): |
| 379 |