Python API#

• Cluster
  • KMeans
    • KMeansParams
    • fit()
    • cluster_cost()
    • compute_new_centroids()
• Common
  • Basic Vocabulary
    • DeviceResources
      • DeviceResources.getHandle()
      • DeviceResources.sync()
    • Stream
      • Stream.get_ptr()
      • Stream.sync()
    • device_ndarray
      • device_ndarray.c_contiguous
      • device_ndarray.copy_to_host()
      • device_ndarray.dtype
      • device_ndarray.empty()
      • device_ndarray.f_contiguous
      • device_ndarray.shape
      • device_ndarray.strides
  • Interruptible
    • cuda_interruptible()
    • synchronize()
    • cuda_yield()
  • CUDA Array Interface Helpers
    • cai_wrapper
• Distance
  • pairwise_distance()
  • fused_l2_nn_argmin()
• Matrix
  • select_k()
• Neighbors
  • Brute Force
    • knn()
  • CAGRA
    • IndexParams
    • build()
    • SearchParams
    • search()
    • Serializer Methods
      • save()
      • load()
  • HNSW
    • SearchParams
    • from_cagra()
    • search()
    • Serializer Methods
      • save()
      • load()
  • IVF-Flat
    • IndexParams
    • build()
    • extend()
    • SearchParams
    • search()
    • Serializer Methods
      • save()
      • load()
  • IVF-PQ
    • IndexParams
    • build()
    • extend()
    • SearchParams
    • search()
    • Serializer Methods
      • save()
      • load()
    • Candidate Refinement
      • refine()
• Random
  • rmat()