SetShapeInfo
产品支持情况
- Ascend 950PR/Ascend 950DT：支持
- Atlas A3 训练系列产品/Atlas A3 推理系列产品：支持
- Atlas A2 训练系列产品/Atlas A2 推理系列产品：支持
- Atlas 200I/500 A2 推理产品：支持
- Atlas 推理系列产品AI Core：支持
- Atlas 推理系列产品Vector Core：支持
- Atlas 训练系列产品：支持
功能说明
设置LocalTensor的Shape信息。
函数原型
C++
__aicore__ inline void SetShapeInfo(const ShapeInfo& shapeInfo)
参数说明
表1 参数说明
参数名 | 输入/输出 | 描述 |
|---|---|---|
shapeInfo | 输入 | Shape信息，ShapeInfo结构体类型。 |
返回值说明
无
约束说明
无
调用示例
C++
// 示例设置Tensor的ShapeInfo信息
AscendC::LocalTensor<float> maxUb = softmaxMaxBuf.template Get<float>();
uint32_t shapeArray[] = {16, 1024};
maxUb.SetShapeInfo(AscendC::ShapeInfo(2, shapeArray, AscendC::DataFormat::ND));