MakeShape

Applicability

Product

Supported

Atlas 350 Accelerator Card

x

Atlas A3 training product/Atlas A3 inference product

Atlas A2 training product/Atlas A2 inference product

Atlas 200I/500 A2 inference product

x

Atlas inference product AI Core

x

Atlas inference product Vector Core

x

Atlas training product

x

Function Usage

Packages the input data into the Shape data structure.

Prototype

1
2
template <typename... Ts>
__aicore__ inline constexpr Shape<Ts...> MakeShape(const Ts&... t)

Parameters

Parameter

Input/Output

Description

Ts...

Input

Formal parameter package of the input type. The usage and restrictions are the same as those of Std::tuple.

Returns

Shape structure type (alias of the Std::tuple type), which is used to define the logical shape of data, such as the number of rows and columns in a two-dimensional matrix or the size of each dimension in a multi-dimensional tensor. The definition is as follows:

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

Restrictions

Same as Std::tuple.

Examples

See Examples.