昇腾社区首页
中文
注册

ReductionConfig

API定义

inline ReductionConfig(std::vector<int> deviceList, const char *model, uint64_t modelSize)

功能描述

ReductionConfig的默认构造函数,当用户使用“NN”降维时,使用该函数。

输入

std::vector<int> deviceList:NPU侧资源配置。

const char *model:神经网络降维模型。

uint64_t modelSize:模型的大小。

输出

返回值

约束说明

使用不同的降维算法,需要配置对应的参数并且降维后的维度需要满足后续使用降维数据index的维度限制

使用NN降维时,应确保模型路径存在,模型使用正确,model的长度应该等于“modelSize”且模型大小不应该大于128MB。