(grid_coord: torch.Tensor, depth: int = 16)
| 39 | |
| 40 | |
| 41 | def z_order_encode(grid_coord: torch.Tensor, depth: int = 16): |
| 42 | x, y, z = grid_coord[:, 0].long(), grid_coord[:, 1].long(), grid_coord[:, 2].long() |
| 43 | # we block the support to batch, maintain batched code in Point class |
| 44 | code = z_order_encode_(x, y, z, b=None, depth=depth) |
| 45 | return code |
| 46 | |
| 47 | |
| 48 | def z_order_decode(code: torch.Tensor, depth): |