(code, depth=16, order="z")
| 26 | |
| 27 | @torch.inference_mode() |
| 28 | def decode(code, depth=16, order="z"): |
| 29 | assert order in {"z", "hilbert"} |
| 30 | batch = code >> depth * 3 |
| 31 | code = code & ((1 << depth * 3) - 1) |
| 32 | if order == "z": |
| 33 | grid_coord = z_order_decode(code, depth=depth) |
| 34 | elif order == "hilbert": |
| 35 | grid_coord = hilbert_decode(code, depth=depth) |
| 36 | else: |
| 37 | raise NotImplementedError |
| 38 | return grid_coord, batch |
| 39 | |
| 40 | |
| 41 | def z_order_encode(grid_coord: torch.Tensor, depth: int = 16): |
nothing calls this directly
no test coverage detected