Python coordinate_map_manager_test Examples

Example #1

File: convolution_cpu.py Project: BestSonny/MinkowskiEngineM

 def test_stride(self):
     coordinates = torch.IntTensor([[0, 1], [1, 2], [2, 3], [2, 3]])
     key, manager, map_inverse_map = _C.coordinate_map_manager_test(coordinates, "")
     unique_map, inverse_map = map_inverse_map
     stride = [2]
     key = _C.coordinate_map_manager_stride(manager, key, stride)
     print(key)

Example #2

 def test(self):
     coordinates = torch.IntTensor([[0, 1], [1, 2], [2, 3], [2, 3]]).to(0)
     key, manager, map_inverse_map = _C.coordinate_map_manager_test(coordinates, "")
     unique_map, inverse_map = map_inverse_map
     self.assertTrue(
         torch.all(coordinates[unique_map.long()][inverse_map.long()] == coordinates)
     )