CANN/metadef:Tensor创建API文档

张开发
2026/5/9 22:40:35 15 分钟阅读

分享文章

CANN/metadef:Tensor创建API文档
CreateFollowing【免费下载链接】metadefAscend Metadata Definition项目地址: https://gitcode.com/cann/metadef函数功能创建一个指定数据类型以及大小的Tensor其数据在TensorV2对象后连续排布。函数原型传入元素个数和数据类型创建Tensorstatic std::unique_ptruint8_t[] CreateFollowing(const int64_t shape_size, const ge::DataType dt, size_t total_size)传入数据类型和Tensor长度创建Tensorstatic std::unique_ptruint8_t[] CreateFollowing(const ge::DataType dt, const size_t tensor_size, size_t total_size)参数说明表 1参数说明传入元素个数和数据类型创建Tensor参数输入/输出说明shape_size输入元素个数。dt输入数据类型DataType类型。total_size输出创建出的Tensor在内存中的长度。包含Tensor对象的长度和Tensor数据的长度。表 2参数说明传入数据类型和Tensor长度创建Tensor参数输入/输出说明dt输入数据类型DataType类型。tensor_size输入Tensor数据的长度。单位为字节。total_size输出创建出的Tensor在内存中的长度。和tensor_size参数不同total_size包含Tensor对象的长度和Tensor数据的长度。单位为字节。返回值说明创建的Tensor指针。约束说明无。调用示例size_t total_size; auto tensor_holder TensorV2::CreateFollowing(32, ge::DataType::DT_INT8, total_size);【免费下载链接】metadefAscend Metadata Definition项目地址: https://gitcode.com/cann/metadef创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考

更多文章