注意

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

矩阵归约

矩阵范数

#include <raft/matrix/norm.cuh>

namespace raft::matrix

template<typename m_t, typename idx_t>
m_t l2_norm(raft::resources const &handle, raft::device_mdspan<const m_t, idx_t> in)

获取矩阵的 L2/F-范数。

参数：

• handle – [输入] raft 句柄

• in – [输入] 总共有 size 个元素的输入矩阵/向量

返回：

矩阵 L2 范数