minuet.nn.functional.voxelize#

Functions

voxelize(coordinates[, voxel_size, dtype, ...])

Voxelize the coordinates with the given voxel_size

voxelize(coordinates: Tensor, voxel_size: float | Tuple[float, ...] = 1, dtype: dtype = torch.int32, return_reverse_indices: bool = False)#

Voxelize the coordinates with the given voxel_size

Parameters:
  • coordinates – the coordinate tensor to be voxelized

  • voxel_size – the size of each voxel

  • dtype – the data type of for the output coordinate tensor

  • return_reverse_indices – whether to return the reverse indices tensor

Returns:

the voxelized coordinates and (optionally) the reverse indices tensor if return_reverse_indices is True