注意

RAFT 中的向量搜索和聚类算法正在迁移到一个专门用于向量搜索的新库 cuVS。在此迁移期间，我们将继续支持 RAFT 中的向量搜索算法，但在 RAPIDS 24.06（六月）发布后将不再更新它们。我们计划在 RAPIDS 24.10（十月）发布前完成迁移，并在 24.12（十二月）发布中将它们从 RAFT 中彻底移除。

C++ API

• 核心
  • 资源
    • 词汇表
      • resource_type
      • raft::resource::resource
      • raft::resource::empty_resource
      • raft::resource::resource_factory
      • raft::resource::empty_resource_factory
    • 设备资源
      • raft::device_resources
    • 设备资源管理器
    • 资源函数
      • 通信
      • cuBLAS 句柄
      • cuBLASLt 句柄
      • CUDA 流
      • CUDA 流池
      • cuSolverDn 句柄
      • cuSolverSp 句柄
      • cuSparse 句柄
      • 设备 ID
      • 设备内存资源
      • 设备属性
      • 子通信器
      • Thrust 执行策略
      • 自定义运行时共享资源
  • 日志
  • 键值对
    • raft::KeyValuePair
      • 键
      • 值
      • KeyValuePair()
      • KeyValuePair()
      • key
      • value
  • NVTX
    • push_range()
    • pop_range()
    • raft::common::nvtx::range
      • range()
  • 可中断
    • raft::interrupted_exception
    • raft::interruptible
      • cancel()
      • synchronize()
      • synchronize()
      • yield()
      • yield_no_throw()
      • get_token()
      • get_token()
      • cancel()
  • 运算符和函数对象
    • add_const_op
    • sub_const_op
    • mul_const_op
    • div_const_op
    • div_checkzero_const_op
    • pow_const_op
    • mod_const_op
    • equal_const_op
    • absdiff_op
    • sqdiff_op
    • raft::identity_op
    • raft::void_op
    • raft::cast_op
    • raft::key_op
    • raft::value_op
    • raft::sqrt_op
    • raft::nz_op
    • raft::abs_op
    • raft::sq_op
    • raft::add_op
    • raft::sub_op
    • raft::mul_op
    • raft::div_op
    • raft::div_checkzero_op
    • raft::pow_op
    • raft::mod_op
    • raft::min_op
    • raft::max_op
    • raft::argmin_op
    • raft::argmax_op
    • raft::greater_op
    • raft::less_op
    • raft::greater_or_equal_op
    • raft::less_or_equal_op
    • raft::equal_op
    • raft::notequal_op
    • raft::const_op
    • raft::plug_const_op
    • raft::compose_op
    • raft::map_args_op
  • 数学函数
    • abs()
  • 位集合
    • raft::core::bitset_view
      • bitset_view()
      • bitset_view()
      • test()
      • operator[]()
      • data()
      • size()
      • n_elements()
      • count()
      • count()
      • repeat()
      • sparsity()
      • get_original_nbits()
      • to_csr()
      • eval_n_elements()
    • raft::core::bitset
      • bitset()
      • bitset()
      • view()
      • data()
      • size()
      • n_elements()
      • to_mdspan()
      • resize()
      • test()
      • set()
      • flip()
      • reset()
      • count()
      • count()
      • any()
      • all()
      • none()
  • 位图
    • raft::core::bitmap_view
      • bitmap_view()
      • bitmap_view()
      • to_csr()
• 线性代数
  • 算术
    • 加法
      • add()
      • add_scalar()
      • add_scalar()
    • 二元运算符
      • binary_op()
    • 除法
      • divide_scalar()
    • 乘法
      • multiply_scalar()
    • 幂
      • power()
      • power_scalar()
    • 平方根
      • sqrt()
    • 减法
      • subtract()
      • subtract_scalar()
      • subtract_scalar()
    • 三元运算符
      • ternary_op()
    • 一元运算符
      • unary_op()
      • write_only_unary_op()
  • BLAS 例程
    • axpy
      • axpy()
      • axpy()
    • 点积
      • dot()
      • dot()
    • gemm
      • gemm()
    • gemv
      • gemv()
  • 映射与归约
    • 合并归约
      • coalesced_reduction()
    • 映射
      • map()
      • map()
      • map()
      • map()
      • map_offset()
      • map_offset()
      • map_offset()
      • map_offset()
    • 映射归约
      • map_reduce()
    • 均方误差
      • mean_squared_error()
    • 范数
      • norm()
    • 归一化
      • row_normalize()
      • row_normalize()
    • 归约
      • reduce()
    • 按键归约列
      • reduce_cols_by_key()
    • 按键归约行
      • reduce_rows_by_key()
    • 带步长归约
      • strided_reduction()
  • 矩阵操作
    • 转置
      • transpose()
  • 矩阵-向量操作
    • 算术
      • binary_mult_skip_zero()
      • binary_div()
      • binary_div_skip_zero()
      • binary_add()
      • binary_sub()
    • 操作
      • matrix_vector_op()
      • matrix_vector_op()
  • 线性代数求解器
    • 特征分解
      • eig_dc()
      • eig_dc_selective()
      • eig_jacobi()
    • QR 分解
      • qr_get_q()
      • qr_get_qr()
    • 随机奇异值分解
      • rsvd_fixed_rank()
      • rsvd_fixed_rank()
      • rsvd_fixed_rank_symmetric()
      • rsvd_fixed_rank_symmetric()
      • rsvd_fixed_rank_jacobi()
      • rsvd_fixed_rank_jacobi()
      • rsvd_fixed_rank_symmetric_jacobi()
      • rsvd_fixed_rank_symmetric_jacobi()
      • rsvd_perc()
      • rsvd_perc()
      • rsvd_perc_symmetric()
      • rsvd_perc_symmetric()
      • rsvd_perc_jacobi()
      • rsvd_perc_jacobi()
      • rsvd_perc_symmetric_jacobi()
      • rsvd_perc_symmetric_jacobi()
      • randomized_svd()
      • randomized_svd()
    • 奇异值分解
      • svd_qr()
      • svd_qr()
      • svd_qr_transpose_right_vec()
      • svd_qr_transpose_right_vec()
      • svd_eig()
      • svd_eig()
      • svd_reconstruction()
    • 最小二乘
      • lstsq_svd_qr()
      • lstsq_svd_jacobi()
      • lstsq_eig()
      • lstsq_qr()
• 矩阵
  • 矩阵算术
    • 按行操作
      • linewise_op()
      • linewise_op()
    • 幂
      • power()
      • weighted_power()
      • weighted_power()
      • power()
    • 比率
      • ratio()
      • ratio()
    • 倒数
      • reciprocal()
      • reciprocal()
    • 符号翻转
      • sign_flip()
    • 平方根
      • sqrt()
      • sqrt()
      • weighted_sqrt()
      • weighted_sqrt()
  • 矩阵操作
    • 初始化
      • fill()
      • fill()
    • 反转
      • col_reverse()
      • row_reverse()
    • 阈值
      • zero_small_values()
      • zero_small_values()
  • 矩阵排序
    • Argmax
      • argmax()
    • Argmin
      • argmin()
    • 选择 K
      • SelectAlgo
      • select_k()
      • operator<<()
      • select_k()
    • 按列排序
      • sort_cols_per_row()
      • sort_cols_per_row()
  • 矩阵归约
    • 矩阵范数
      • l2_norm()
• 多维数据
  • 多维表示
    • 数据布局
      • layout_c_contiguous
      • row_major
      • layout_f_contiguous
      • col_major
    • 形状
      • matrix_extent
      • vector_extent
      • scalar_extent
      • extent_3d
      • extent_4d
      • extent_5d
      • flatten()
      • reshape()
      • accessor_of_const()
      • accessor_of_const()
      • flatten()
      • reshape()
    • 访问器
      • raft::host_device_accessor
      • host_accessor
      • device_accessor
      • managed_accessor
  • mdspan: 多维非拥有视图
    • mdspan
    • make_mdspan()
    • make_extents()
    • make_strided_layout()
    • unravel_index()
    • is_c_contiguous()
    • is_f_contiguous()
    • make_const_mdspan()
    • 设备词汇表
      • device_mdspan
      • raft::is_device_mdspan
      • is_device_mdspan_t
      • is_input_device_mdspan_t
      • is_output_device_mdspan_t
      • enable_if_device_mdspan
      • enable_if_input_device_mdspan
      • enable_if_output_device_mdspan
      • device_matrix_view
      • device_vector_view
      • device_scalar_view
    • 设备工厂
      • make_device_matrix_view()
      • make_device_vector_view()
      • make_device_scalar_view()
    • Managed 词汇表
      • managed_mdspan
      • raft::is_managed_mdspan
      • is_managed_mdspan_t
      • is_input_managed_mdspan_t
      • is_output_managed_mdspan_t
      • enable_if_managed_mdspan
      • enable_if_input_managed_mdspan
      • enable_if_output_managed_mdspan
    • Managed 工厂
      • make_managed_mdspan()
    • 主机词汇表
      • host_mdspan
      • raft::is_host_mdspan
      • is_host_mdspan_t
      • is_input_host_mdspan_t
      • is_output_host_mdspan_t
      • enable_if_host_mdspan
      • enable_if_input_host_mdspan
      • enable_if_output_host_mdspan
      • host_matrix_view
      • host_vector_view
      • host_scalar_view
    • 主机工厂
      • make_host_matrix_view()
      • make_host_vector_view()
      • make_host_scalar_view()
    • 固定内存词汇表
      • pinned_mdspan
      • pinned_matrix_view
      • pinned_vector_view
      • pinned_scalar_view
    • 固定内存工厂
      • make_pinned_matrix_view()
      • make_pinned_scalar_view()
    • 验证例程
      • raft::is_mdspan
      • is_mdspan_t
      • raft::is_input_mdspan
      • is_input_mdspan_t
      • raft::is_output_mdspan
      • is_output_mdspan_t
      • enable_if_mdspan
      • enable_if_input_mdspan
      • enable_if_output_mdspan
  • mdarray: 多维拥有容器
    • is_array_interface_v
    • raft::array_interface
    • raft::is_array_interface
    • raft::is_array_interface< T1 >
    • raft::is_array_interface< T1, Tn... >
    • raft::mdarray
      • view_type
      • mdarray()
      • view()
      • view()
      • operator()()
      • operator()()
    • 设备词汇表
      • device_mdarray
      • device_matrix
      • device_vector
      • device_scalar
    • 设备工厂
      • make_device_matrix()
      • make_device_vector()
      • make_device_scalar()
    • 主机词汇表
      • host_matrix
      • host_vector
      • host_scalar
    • 主机工厂
      • make_host_mdarray()
      • make_host_matrix()
      • make_host_scalar()
      • make_host_vector()
    • Managed 词汇表
      • managed_mdarray
      • managed_matrix
      • managed_vector
      • managed_scalar
    • Managed 工厂
      • make_managed_matrix()
      • make_managed_vector()
      • make_managed_scalar()
    • 固定内存词汇表
      • pinned_mdarray
      • pinned_matrix
      • pinned_vector
      • pinned_scalar
    • 固定内存工厂
      • make_pinned_matrix()
      • make_pinned_vector()
      • make_pinned_scalar()
  • span: 一维非拥有视图
    • element_type
    • value_type
    • size_type
    • difference_type
    • pointer
    • const_pointer
    • reference
    • const_reference
    • iterator
    • const_iterator
    • reverse_iterator
    • const_reverse_iterator
    • span()
    • span()
    • span()
    • span()
    • span()
    • span()
    • span()
    • begin()
    • end()
    • cbegin()
    • cend()
    • front()
    • back()
    • operator[]()
    • data()
    • size()
    • size_bytes()
    • empty()
    • first()
    • first()
    • last()
    • last()
    • subspan()
    • subspan()
    • operator==()
    • operator!=()
    • operator<()
    • operator<=()
    • operator>()
    • operator>=()
    • as_bytes()
    • as_writable_bytes()
    • raft::span
    • device_span
    • host_span
  • mdbuffer: 多维可能拥有容器
    • alternate_from_mem_type
    • default_container_policy_variant
    • is_mdbuffer_t
    • is_input_mdbuffer_t
    • is_output_mdbuffer_t
    • enable_if_mdbuffer
    • enable_if_input_mdbuffer
    • enable_if_output_mdbuffer
    • is_mdbuffer_v
    • is_input_mdbuffer_v
    • is_output_mdbuffer_v
    • variant_index_from_memory_type()
    • memory_type_from_variant_index()
    • __takes_an_mdbuffer_ptr()
    • raft::default_buffer_container_policy
    • raft::mdbuffer
      • mdbuffer()
      • mdbuffer()
      • mdbuffer()
      • mdbuffer()
      • mdbuffer()
      • mdbuffer()
      • mdbuffer()
      • mdbuffer()
      • mem_type()
      • is_owning()
      • view()
      • view()
      • view()
      • view()
    • raft::is_mdbuffer
    • raft::is_mdbuffer< T, std::void_t< decltype(__takes_an_mdbuffer_ptr(std::declval< T * >()))> >
    • raft::is_input_mdbuffer
    • raft::is_input_mdbuffer< T, std::void_t< decltype(__takes_an_mdbuffer_ptr(std::declval< T * >()))> >
    • raft::is_output_mdbuffer
    • raft::is_output_mdbuffer< T, std::void_t< decltype(__takes_an_mdbuffer_ptr(std::declval< T * >()))> >
  • memory_type_dispatcher
    • memory_type_dispatcher()
    • memory_type_dispatcher()
    • memory_type_dispatcher()
  • temporary_device_buffer: 临时 raft::device_mdspan 生成对象
    • raft::temporary_device_buffer
      • temporary_device_buffer()
      • view()
    • 工厂
      • make_temporary_device_buffer()
      • make_readonly_temporary_device_buffer()
      • make_writeback_temporary_device_buffer()
• 多节点多 GPU
  • 常见类型
    • datatype_t
      • CHAR
      • UINT8
      • INT32
      • UINT32
      • INT64
      • UINT64
      • FLOAT32
      • FLOAT64
    • op_t
      • SUM
      • PROD
      • MIN
      • MAX
    • status_t
      • SUCCESS
      • ERROR
      • ABORT
    • request_t
    • get_type()
    • get_type< char >()
    • get_type< uint8_t >()
    • get_type< int >()
    • get_type< uint32_t >()
    • get_type< int64_t >()
    • get_type< uint64_t >()
    • get_type< float >()
    • get_type< double >()
  • 通信接口
    • raft::comms::comms_t
      • ~comms_t()
      • get_size()
      • get_rank()
      • comm_split()
      • barrier()
      • sync_stream()
      • isend()
      • irecv()
      • waitall()
      • allreduce()
      • bcast()
      • bcast()
      • reduce()
      • allgather()
      • allgatherv()
      • gather()
      • gatherv()
      • reducescatter()
      • device_send()
      • device_recv()
      • device_sendrecv()
      • device_multicast_sendrecv()
      • group_start()
      • group_end()
  • MPI 通信
    • initialize_mpi_comms()
  • NCCL+UCX 通信
    • build_comms_nccl_only()
    • build_comms_nccl_ucx()
• 随机
  • 随机状态
    • raft::random::RngState
      • type
    • 数据生成
      • make_blobs
      • make_regression
      • rmat
    • 单变量随机抽样
      • uniform()
      • uniformInt()
      • normal()
      • normalInt()
      • normalTable()
      • fill()
      • bernoulli()
      • scaled_bernoulli()
      • gumbel()
      • lognormal()
      • logistic()
      • exponential()
      • rayleigh()
      • laplace()
      • discrete()
    • 多变量随机抽样
      • 多变量高斯
    • 无放回抽样
      • sample_without_replacement()
      • sample_without_replacement()
      • permute()
      • permute()
• 求解器
  • 线性分配问题
    • raft::solver::LinearAssignmentProblem
      • LinearAssignmentProblem()
      • solve()
      • getRowDualVector()
      • getColDualVector()
      • getPrimalObjectiveValue()
      • getDualObjectiveValue()
  • 最小生成树
    • mst()
• 稀疏
  • 稀疏类型
    • SparsityType
      • OWNING
      • PRESERVING
    • row_type
    • col_type
    • nnz_type
    • element_type
    • structure_view_type
    • view_type
    • structure_type
    • container_policy_type
    • container_type
    • sparse_structure()
    • sparse_structure()
    • get_nnz()
    • get_n_rows()
    • get_n_cols()
    • initialize_sparsity()
    • sparse_matrix_view()
    • structure_view()
    • get_elements()
    • sparse_matrix()
    • sparse_matrix()
    • sparse_matrix()
    • sparse_matrix()
    • ~sparse_matrix()
    • view()
    • raft::sparse_structure
    • raft::sparse_matrix_view
    • raft::sparse_matrix
    • COO 矩阵
      • 基本词汇表
      • 设备 COO 矩阵
      • 主机 COO 矩阵
    • CSR 矩阵
      • 基本词汇表
      • 设备 CSR 矩阵
      • 主机 CSR 矩阵
  • 稀疏线性代数
    • csr_add_calc_inds()
    • csr_add_finalize()
    • coo_degree()
    • coo_degree()
    • coo_degree_scalar()
    • coo_degree_scalar()
    • coo_degree_nz()
    • coo_degree_nz()
    • compute_graph_laplacian()
    • csr_row_normalize_l1()
    • csr_row_normalize_max()
    • rowNormCsr()
    • sddmm()
    • spmm()
    • coo_symmetrize()
    • symmetrize()
    • csr_transpose()
  • 稀疏矩阵操作
    • coo_remove_scalar()
    • coo_remove_scalar()
    • coo_remove_zeros()
    • compute_duplicates_mask()
    • max_duplicates()
    • csr_row_slice_indptr()
    • csr_row_slice_populate()
    • coo_sort()
    • coo_sort()
    • coo_sort_by_weight()
  • 稀疏求解器
    • lanczos_compute_smallest_eigenvectors()
    • lanczos_compute_smallest_eigenvectors()
    • mst()
    • raft::sparse::solver::Graph_COO
    • raft::sparse::solver::lanczos_solver_config
      • n_components
      • max_iterations
      • ncv
      • tolerance
      • seed
    • raft::sparse::solver::MST_solver
• 统计
  • 汇总统计
    • 协方差
      • cov()
    • 直方图
      • HistType
      • histogram()
    • 均值
      • mean()
      • mean()
    • 均值中心化
      • mean_center()
      • mean_add()
    • 均值方差
      • meanvar()
    • 最小值/最大值
      • minmax()
    • 标准差
      • stddev()
    • 求和
      • sum()
    • 加权平均
      • weighted_mean()
      • row_weighted_mean()
      • col_weighted_mean()
  • 概率与信息论
    • 列联表
      • get_input_class_cardinality()
      • contingency_matrix()
    • 熵
      • entropy()
    • KL 散度
      • kl_divergence()
    • 互信息
      • mutual_info_score()
  • 回归模型评分
    • 信息准则
      • IC_Type
      • information_criterion_batched()
    • R2 分数
      • r2_score()
    • 回归指标
      • regression_metrics()
  • 分类模型评分
    • 准确率
      • accuracy()
  • 聚类模型评分
    • 调整兰德指数
      • adjusted_rand_index()
    • 完备性评分
      • completeness_score()
    • 聚类离散度
      • cluster_dispersion()
    • 兰德指数
      • rand_index()
    • 轮廓系数
      • silhouette_score()
      • silhouette_score_batched()
    • V-度量
      • v_measure()
• 工具
  • 内存池