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)templatetypename T aclTensor *ConvertToTensor(const T *value, uint64_t size, DataType dataType)参数说明参数输入/输出说明value输入host侧源数据。size输入源数据的元素个数。dataType输入将源数据转换为指定数据类型后写入aclTensor。返回值说明返回转换后的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创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考
CANN opbase数据转换API
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)templatetypename T aclTensor *ConvertToTensor(const T *value, uint64_t size, DataType dataType)参数说明参数输入/输出说明value输入host侧源数据。size输入源数据的元素个数。dataType输入将源数据转换为指定数据类型后写入aclTensor。返回值说明返回转换后的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创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考