TensorDesc

功能

包含对Tensor的相关描述信息:每个Tensor的数据类型,数据排布格式和形状维度信息。

定义

1
2
3
4
5
struct TensorDesc {
    aclDataType dtype = ACL_DT_UNDEFINED;
    aclFormat format = ACL_FORMAT_UNDEFINED;
    Dims shape;
};

参数列表

成员名称

数据类型

默认值

描述

dtype

aclDataType

ACL_DT_UNDEFINED

Tensor数据类型。

具体请参见aclDataType

Atlas 推理系列产品 中不支持ACL_BF16(bf16)类型数据。

format

aclFormat

ACL_FORMAT_UNDEFINED

Tensor数据排布格式。

具体请参见aclFormat

shape

Dims

-

Tensor数据的形状,Tensor的维度信息须满足Dims的参数约束要求,具体请参见Dims