GetStride

产品支持情况

产品	是否支持
Ascend 950PR/Ascend 950DT	√

功能说明

获取描述内存访问步长的Stride对象。

函数原型

template <size_t... Is, typename ShapeType, typename StrideType>
__aicore__ inline constexpr auto GetStride(const Layout<ShapeType, StrideType>& layout)

参数说明

表 1 模板参数说明

参数名	输入/输出	描述
Is...	输入	可选。多级索引递归选取。
ShapeType	输入	Layout的shape类型。
StrideType	输入	Layout的stride类型。

表 2 参数说明

参数名	输入/输出	描述
layout	输入	Layout对象。

返回值说明

描述内存访问步长的Stride对象。

约束说明

• layout必须是有效的Layout对象。
• 索引Is...必须在有效范围内。

调用示例

using namespace AscendC::Te;

auto layout = MakeNDLayout(128,128);
// 无模板实参时Is... 为空，等价于layout.Stride() 得到整个stride元组
auto strideTuple = GetStride(layout);

// 带索引时等价于layout.Stride<0>()，按递归取stride子结构或某一维
auto stride0 = GetStride<0>(layout);