(self, hidden_states)
| 295 | |
| 296 | |
| 297 | def patchify(self, hidden_states): |
| 298 | hidden_states = rearrange(hidden_states, "B C (H P) (W Q) -> B (H W) (C P Q)", P=2, Q=2) |
| 299 | return hidden_states |
| 300 | |
| 301 | |
| 302 | def unpatchify(self, hidden_states, height, width): |
no outgoing calls
no test coverage detected