GetShape

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	x
Atlas inference product's AI Core	x
Atlas inference product's Vector Core	x
Atlas training products	x

Function

Returns the Shape object that describes the tensor shape.

Prototype

__aicore__ inline constexpr decltype(auto) GetShape() {}   
__aicore__ inline constexpr decltype(auto) GetShape() const {}

Parameters

None

Returns

Shape object that describes the tensor shape. The structure type (alias of the Std::tuple type) is defined as follows:

template <typename... Shapes>
using Shape = Std::tuple<Shapes...>;

Restrictions

None

Example

// Initialize the Layout data structure and obtain the corresponding value.
AscendC::Shape<int,int,int> shape = AscendC::MakeShape(10, 20, 30);
AscendC::Stride<int,int,int> stride = AscendC::MakeStride(1, 100, 200);

auto layoutMake = AscendC::MakeLayout(shape, stride);
AscendC::Layout<AscendC::Shape<int, int, int>, AscendC::Stride<int, int, int>> layoutInit(shape, stride);

int value = AscendC::Std::get<0>(layoutMake.GetShape()); // value = 10
value = AscendC::Std::get<1>(layoutMake.GetShape()); // value = 20
value = AscendC::Std::get<2>(layoutMake.GetShape()); // value = 30

Parent topic: Layout