更新时间:2021-03-18 GMT+08:00
分享

AIModelManager::CreateInputTensor

创建输入的Tensor列表。

如果用户使用该接口来创建tensor,该接口返回的内存首地址满足512对齐要求。建议用户使用AITensorFactory::CreateTensor接口来创建tensor,详见AITensorFactory::CreateTensor

函数格式

virtual AIStatus AIModelManager::CreateInputTensor(std::vector<std::shared_ptr<IAITensor>> &in_data);

参数说明

参数

说明

取值范围

in_data

输入tensor列表。

关于IAITensor数据类型的定义,请参见IAITensor

-

返回值

SUCCESS初始化成功/FAILED初始化失败。

分享:

    相关文档

    相关产品