(window_size, window_shift, x)
| 285 | |
| 286 | |
| 287 | def shifted_window(window_size, window_shift, x): |
| 288 | x = torch.roll(x, shifts=(window_shift, window_shift), dims=(-2, -3)) |
| 289 | windows = window(window_size, x) |
| 290 | return windows |
| 291 | |
| 292 | |
| 293 | def shifted_unwindow(window_shift, x): |
no test coverage detected