文档
注册

函数:create_tensor_desc

C函数原型

aclTensorDesc *aclCreateTensorDesc(aclDataType dataType, int numDims, const int64_t *dims, aclFormat format)

Python函数

tensor_desc = acl.create_tensor_desc(data_type, dims, format)

函数功能

创建aclTensorDesc类型的数据,该数据类型用于描述tensor的数据类型、shape、format等信息。同步接口。

如需销毁aclTensorDesc类型的数据,请参见acl.destroy_tensor_desc

输入说明

data_type:int,表述tensor描述的数据类型。

dims:int list,tensor描述指定维度的大小。dims是一个int的list,数组中的每个元素表示tensor中每个维度的大小,如果数组中某个元素的值为0,则为空tensor。

如果用户需要使用空tensor,则在申请内存时,内存大小最小为1Byte,以保障后续业务正常运行。

format:int,tensor描述的format。

返回值说明

tensor_desc:int,创建tensor描述的地址对象。

约束说明

注意事项

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词