CANN/opbase分配张量API

AllocTensor

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

功能说明

申请一个device侧tensor，提供多个重载函数，可以指定不同的属性。

函数原型

aclTensor *AllocTensor(const Shape &shape, DataType dataType, Format format = FORMAT_ND)

aclTensor *AllocTensor(const Shape &storageShape, const Shape &originShape, DataType dataType, Format storageFormat, Format originFormat)

aclTensor *AllocTensor(DataType dataType, Format storageFormat, Format originFormat)

参数说明

返回值说明

返回申请得到的aclTensor，申请失败则返回nullptr。

约束说明

入参指针不能为空。

调用示例

// 申请一个int64类型，shape为[1, 2, 3, 4, 5]的ND tensor void Func(aclOpExecutor *executor) { gert::Shape newShape; for (int64_t i = 1; i <= 5; i++) { newShape.AppendDim(i); } aclTensor *tensor = executor->AllocTensor(newShape, DT_INT64, ge::FORMAT_ND); }

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