SetShapeInfo

Product Support

Product

Supported

Atlas A3 training products/Atlas A3 inference products

√

Atlas A2 training products/Atlas A2 inference products

√

Atlas 200I/500 A2 inference products

√

Atlas inference product's AI Core

√

Atlas inference product's Vector Core

√

Atlas training products

√

Function

Sets the shape information of a LocalTensor.

Prototype

1
__aicore__ inline void SetShapeInfo(const ShapeInfo& shapeInfo)

Parameters

Table 1 Parameters

Parameter

Input/Output

Description

shapeInfo

Input

Shape information, which is of the ShapeInfo structure type.

Returns

None

Restrictions

None

Example

For details, see Example.