昇腾社区首页
中文
注册
开发者
下载

SetStride

函数功能

设置指定idx轴的步长值。

函数原型

1
void SetStride(size_t idx, const int64_t stride)

参数说明

参数

输入/输出

说明

idx

输入

维度的索引,调用者需要保证索引合法。

stride

输入

对idx轴设置的步长。

返回值说明

无。

约束说明

调用者需要保证index合法。

调用示例

1
2
Stride stride0({3, 256, 256});
stride0.SetStride(0U, 1); // 1,256,256