(self, point, order)
| 371 | |
| 372 | @torch.no_grad() |
| 373 | def get_rel_pos(self, point, order): |
| 374 | K = self.patch_size |
| 375 | rel_pos_key = f"rel_pos_{self.order_index}" |
| 376 | if rel_pos_key not in point.keys(): |
| 377 | grid_coord = point.grid_coord[order] |
| 378 | grid_coord = grid_coord.reshape(-1, K, 3) |
| 379 | point[rel_pos_key] = grid_coord.unsqueeze(2) - grid_coord.unsqueeze(1) |
| 380 | return point[rel_pos_key] |
| 381 | |
| 382 | @torch.no_grad() |
| 383 | def get_padding_and_inverse(self, point): |