企业官网建设流程全解析

ConvertToTensor

【免费下载链接】opbase本项目是CANN算子库的基础框架库，为算子提供公共依赖文件和基础调度能力。项目地址: https://gitcode.com/cann/opbase

功能说明

将不同数据类型的host侧数据，转为一个host侧aclTensor对象。

函数原型

aclTensor *ConvertToTensor(const aclIntArray *value, DataType dataType)

aclTensor *ConvertToTensor(const aclBoolArray *value, DataType dataType)

aclTensor *ConvertToTensor(const aclFloatArray *value, DataType dataType)

aclTensor *ConvertToTensor(const aclFp16Array *value, DataType dataType)

aclTensor *ConvertToTensor(const aclBf16Array *value, DataType dataType)

aclTensor *ConvertToTensor(const aclScalar *value, DataType dataType)

template<typename T> aclTensor *ConvertToTensor(const T *value, uint64_t size, DataType dataType)

参数说明

返回值说明

返回转换后的host侧tensor。

约束说明

入参指针不能为空。

调用示例

// 分别将一个aclScalar和int64_t转为host侧tensor void Func(aclOpExecutor *executor) { int64_t val = 5; aclScalar *scalar = executor->AllocScalar(val); aclTensor *tensor = executor->ConvertToTensor(scalar, DT_INT64); tensor = executor->ConvertToTensor(&val, 1, DT_INT64); }

【免费下载链接】opbase本项目是CANN算子库的基础框架库，为算子提供公共依赖文件和基础调度能力。项目地址: https://gitcode.com/cann/opbase