SetShapeInfo
产品支持情况
产品 |
是否支持 |
|---|---|
Atlas 350 加速卡 |
√ |
√ |
|
√ |
|
√ |
|
√ |
|
√ |
|
√ |
功能说明
设置GlobalTensor的shape信息。
函数原型
1 | __aicore__ inline void SetShapeInfo(const ShapeInfo& shapeInfo) |
参数说明
参数名 |
输入/输出 |
描述 |
|---|---|---|
shapeInfo |
输入 |
ShapeInfo结构体。 |
返回值说明
无。
约束说明
无。
调用示例
1 2 3 4 | // 示例 设置Tensor的ShapeInfo信息 AscendC::GlobalTensor<float> tglobal; uint32_t shapeArray[] = {16, 1024}; tglobal.SetShapeInfo(AscendC::ShapeInfo(2, shapeArray, AscendC::DataFormat::ND)); |
父主题： GlobalTensor