GetShapeInfo

产品支持情况

产品

是否支持

Atlas A3 训练系列产品/Atlas A3 推理系列产品

√

Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件

√

Atlas 200I/500 A2 推理产品

√

Atlas 推理系列产品AI Core

√

Atlas 推理系列产品Vector Core

√

Atlas 训练系列产品

√

Atlas 200/300/500 推理产品

x

功能说明

获取LocalTensor的Shape信息。注意:Shape信息没有默认值,只有通过SetShapeInfo设置过Shape信息后,才可以调用该接口获取正确的Shape信息。

函数原型

1
__aicore__ inline ShapeInfo GetShapeInfo() const

参数说明

无

返回值

LocalTensor的Shape信息,ShapeInfo结构体类型。

约束说明

无

调用示例

参考调用示例。