stride
stride用于定义张量各维度在内存中的步长,用于描述同一维度相邻元素在内存中的间隔。
头文件路径为:tensor_api/tensor.h。
原型定义
template <typename... Strides>
using stride = Std::tuple<Strides...>;
参数说明
表1 参数说明
| 参数名 | 输入/输出 | 描述 |
|---|---|---|
| Strides... | 输入 | 可变参数模板,表示各维度的步长值,也可表示嵌套的stride子结构。 |
API映射关系
stride通常通过make_stride函数创建,通过get_stride函数和layout的stride成员函数获取。