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