aclCreateTensorDesc
产品支持情况
产品
|
是否支持
|
|---|
Atlas A3 训练系列产品/Atlas A3 推理系列产品
|
√
|
Atlas A2 训练系列产品/Atlas A2 推理系列产品
|
√
|
功能说明
创建aclTensorDesc类型的数据，该数据类型用于描述tensor的数据类型、shape、format等信息。
如需销毁aclTensorDesc类型的数据，请参见aclDestroyTensorDesc。
函数原型
aclTensorDesc *aclCreateTensorDesc(aclDataType dataType,
int numDims,
const int64_t *dims,
aclFormat format)
参数说明
参数名
|
输入/输出
|
说明
|
|---|
dataType
|
输入
|
tensor描述的数据类型。
|
numDims
|
输入
|
tensor描述shape的维度个数。
numDims存在以下约束：
- numDims必须大于或等于0；
- numDims>0时，numDims的值必须与dims数组的长度保持一致；
- numDims=0时，系统不使用dims数组值，dims参数值无效。
|
dims
|
输入
|
tensor描述维度大小的指针。
dims是一个数组，数组中的每个元素表示tensor中每个维度的大小，如果数组中某个元素的值为0，则为空tensor。
如果用户需要使用空tensor，则在申请内存时，内存大小最小为1Byte，以保障后续业务正常运行。
|
format
|
输入
|
tensor描述的format。
|
返回值说明
返回aclTensorDesc类型的指针。