下载
EN
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

ShapeInfo

功能说明

ShapeInfo用来存放LocalTensor或GlobalTensor的shape信息。

定义原型

  • ShapeInfo结构定义
     1
     2
     3
     4
     5
     6
     7
     8
     9
    10
    11
    12
    13
    struct ShapeInfo {
    public:
        __aicore__ inline ShapeInfo();
        __aicore__ inline ShapeInfo(const uint8_t inputShapeDim, const uint32_t inputShape[],
            const uint8_t inputOriginalShapeDim, const uint32_t inputOriginalShape[], const DataFormat inputFormat);
        __aicore__ inline ShapeInfo(const uint8_t inputShapeDim, const uint32_t inputShape[], const DataFormat inputFormat);
        __aicore__ inline ShapeInfo(const uint8_t inputShapeDim, const uint32_t inputShape[]);
        uint8_t shapeDim;
        uint8_t originalShapeDim;
        uint32_t shape[K_MAX_DIM];
        uint32_t originalShape[K_MAX_DIM];
        DataFormat dataFormat;
    };
    
  • 获取Shape中所有dim的累乘结果

    __aicore__ inline int GetShapeSize(const ShapeInfo& shapeInfo)

函数说明

表1 ShapeInfo结构参数说明

参数名称

类型

描述

shapeDim

uint8_t

现有的shape维度

shape

uint32_t

现有的shape

originalShapeDim

uint8_t

原始的shape维度

originalShape

uint32_t

原始的shape

dataFormat

DataFormat

数据排布格式
enum DataFormat {     NCHW = 0;     NHWC = 1; }
NCHW:数据按NCHW排布
NHWC:数据按NHWC排布
表2 GetShapeSize参数说明

函数名称

入参说明

含义

shapeInfo

Tensor的shape信息

用来存放LocalTensor或GlobalTensor的shape信息。

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

当前产品无相关内容

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