SetShapeInfo

Applicability

Product	Supported
Atlas 350 Accelerator Card	√
Atlas A3 training product/Atlas A3 inference product	√
Atlas A2 training product/Atlas A2 inference product	√
Atlas 200I/500 A2 inference product	√
Atlas inference product AI Core	√
Atlas inference product Vector Core	√
Atlas training product	√

Function Usage

Sets the shape information of the GlobalTensor.

Prototype

__aicore__ inline void SetShapeInfo(const ShapeInfo& shapeInfo)

Parameters

**Table 1** Parameters
Parameter	Input/Output	Description
shapeInfo	Input	ShapeInfo structure.

Returns

None

Restrictions

None

Examples

// Example: Set ShapeInfo of a tensor.
AscendC::GlobalTensor<float> tglobal;
uint32_t shapeArray[] = {16, 1024};
tglobal.SetShapeInfo(AscendC::ShapeInfo(2, shapeArray, AscendC::DataFormat::ND));

Parent topic: GlobalTensor