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