Select

产品支持情况

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

功能说明

选择Layout的shape和stride指定维度组成新的layout对象并返回。

函数原型

template <size_t... Is, typename Shape, typename Stride>
__aicore__ inline constexpr auto Select(const Layout<Shape, Stride>& layout)

参数说明

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

返回值说明

返回子Layout对象。

约束说明

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

调用示例

using namespace AscendC::Te;

auto shape = MakeShape(10, 20, 30);
auto layout = MakeLayout(shape);

// 选择第0和第1维度
auto subLayout = Select<0, 1>(layout);