昇腾社区首页
中文
注册

train

API定义

void train(idx_t n, const float *x) override

功能描述

对AscendLinearTransform执行训练量化器,继承faiss::VectorTransform中定义的虚函数接口,并提供具体的实现。

输入

Index::idx_t n:训练集中特征向量的条数。

const float *x:特征向量数据。

输出

返回值

约束说明

  • 此处“n”通常不允许超过1e9。
  • 此处指针的“x”需要为非空指针,且长度应该为dim * n,否则可能出现越界读写的错误并引起程序崩溃。