功能说明
设置GlobalTensor的shape信息。
函数原型
| __aicore__ inline void SetShapeInfo(const ShapeInfo& shapeInfo)
|
参数说明
表1 参数说明参数名
|
输入/输出
|
描述
|
|---|
shapeInfo
|
输入
|
ShapeInfo结构体。
|
支持的型号
Atlas 训练系列产品
Atlas 推理系列产品AI Core
Atlas 推理系列产品Vector Core
Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件
Atlas 200I/500 A2 推理产品
Atlas A3 训练系列产品/Atlas A3 推理系列产品