SetShapeInfo
产品支持情况
产品
|
是否支持
|
Ascend 950PR/Ascend 950DT
|
√
|
Atlas A3 训练系列产品/Atlas A3 推理系列产品
|
√
|
Atlas A2 训练系列产品/Atlas A2 推理系列产品
|
√
|
功能说明
设置LocalTensor的Shape信息。
函数原型
__aicore__ inline void SetShapeInfo(const ShapeInfo& shapeInfo)
参数说明
表 1 参数说明
参数名
|
输入/输出
|
描述
|
shapeInfo
|
输入
|
Shape信息,ShapeInfo结构体类型。
|
返回值说明
无
约束说明
无
调用示例
参考调用示例。