函数:get_tensor_desc_dim_v2
C函数原型 |
aclError aclGetTensorDescDimV2(const aclTensorDesc *desc, size_t index, int64_t *dimSize) |
---|---|
Python函数 |
dim_size, ret = acl.get_tensor_desc_dim_v2(desc, index) |
函数功能 |
获取Tensor描述中指定维度的大小。 |
输入说明 |
desc:int,aclTensorDesc类型的指针地址。 需提前调用acl.create_tensor_desc接口创建aclTensorDesc类型。 index:int,指定获取第几个维度的大小,“index”值从0开始。 用户调用acl.get_tensor_desc_num_dims接口获取Shape维度个数,“index”的取值范围:[0, (shape维度个数-1)]。 |
返回值说明 |
dim_size:int,Tensor描述中“index”指定维度的大小。当返回的“dim_size”参数值为“-1”时,表示维度的大小是动态的。 ret:int,错误码。
|
约束说明 |
当acl.get_tensor_desc_num_dims接口的返回值为ACL_UNKNOWN_RANK(0xFFFFFFFFFFFFFFFE)时,表示动态Shape场景下维度个数未知,则不能调用acl.get_tensor_desc_dim_v2接口获取指定维度的范围。 |
父主题: aclmdlTensorDesc