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; };
函数说明
参数名称 |
类型 |
描述 |
---|---|---|
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排布 |
函数名称 |
入参说明 |
含义 |
---|---|---|
shapeInfo |
Tensor的shape信息 |
用来存放LocalTensor或GlobalTensor的shape信息。 |
父主题: 数据类型定义