昇腾社区首页
中文
注册

函数:create_tensor_desc

产品支持情况

产品

是否支持

Atlas A3 训练系列产品/Atlas A3 推理系列产品

Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件

Atlas 训练系列产品

Atlas 推理系列产品

Atlas 200I/500 A2 推理产品

Atlas 200/300/500 推理产品

功能说明

创建aclTensorDesc类型的数据,该数据类型用于描述Tensor的数据类型、Shape、Format等信息。

如需销毁aclTensorDesc类型的数据,请参见函数:destroy_tensor_desc

函数原型

  • C函数原型
    1
    aclTensorDesc *aclCreateTensorDesc(aclDataType dataType, int numDims, const int64_t *dims, aclFormat format)
    
  • python函数
    1
    tensor_desc = acl.create_tensor_desc(data_type, dims, format)
    

参数说明

参数名

说明

data_type

int,表述Tensor描述的数据类型。

dims

list,Tensor描述指定维度大小。“dims”是一个int的列表,列表中的每项表示Tensor中每个维度的大小,如果列表中某个元素的值为0,则为空Tensor。如果用户需要使用空Tensor,则在申请内存时,内存大小最小为1Byte,以保障后续业务正常运行。

format

int,Tensor描述的Format。

返回值说明

返回值

说明

tensor_desc

int,创建Tensor描述的地址对象。