昇腾社区首页
中文
注册

TensorDesc

说明

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

定义

struct TensorDesc {
    aclDataType dtype = ACL_DT_UNDEFINED;
    aclFormat format = ACL_FORMAT_UNDEFINED;
    Dims shape;
};

成员

成员名称

描述

dtype

Tensor数据类型。

具体请参见aclDataType

format

Tensor数据排布格式。

具体请参见aclFormat

shape

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