aclCreateTensorDesc
函数功能
创建aclTensorDesc类型的数据,该数据类型用于描述tensor的数据类型、shape、format等信息。
如需销毁aclTensorDesc类型的数据,请参见aclDestroyTensorDesc。
函数原型
aclTensorDesc *aclCreateTensorDesc(aclDataType dataType,
int numDims,
const int64_t *dims,
aclFormat format)
参数说明
参数名 |
输入/输出 |
说明 |
---|---|---|
dataType |
输入 |
tensor描述的数据类型。 |
numDims |
输入 |
tensor描述shape的维度个数。 numDims存在以下约束:
|
dims |
输入 |
tensor描述维度大小的指针。 dims是一个数组,数组中的每个元素表示tensor中每个维度的大小,如果数组中某个元素的值为0,则为空tensor。 如果用户需要使用空tensor,则在申请内存时,内存大小最小为1Byte,以保障后续业务正常运行。 |
format |
输入 |
tensor描述的format。 |
返回值说明
返回aclTensorDesc类型的指针。
父主题: aclTensorDesc