Usage
=====

The package was designed to be easily used through the experiments.

Here is an example of how you can import it

.. code-block:: python

    import numpy as np
    import torch

    from altametris.pctools.nearest_neighbors import nn_cpu, nn_gpu
    from altametris.pctools.grid_subsampling import cpu as gs

    # create a random point cloud
    pc = np.random.rand(1000, 3)

    # compute knn on CPU
    idx = nn_cpu.knn(pc, pc, K=2)

    # compute knn on GPU
    pc_tensor = torch.from_numpy(pc)
    idx = nn_gpu.knn(pc, pc, K=2)

    # subsample
    subpoints, _, _ = gs.subsample(points, grid_size=0.01)