MakeStride

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 Stride data structure.

Prototype

1
2
template <typename... Ts>
__aicore__ inline constexpr Stride<Ts...> MakeStride(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

Stride structure type (alias of the Std::tuple type), which is used to define the stride of each dimension in memory, that is, the interval between adjacent elements of the same dimension in memory. The stride corresponds to the dimension information of the shape. The definition is as follows:

1
2
template <typename... Strides>
using Stride = Std::tuple<Strides...>;

Restrictions

Same as Std::tuple.

Examples

See Examples.