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

operator[]

函数功能

获取指定idx轴的步长值。

函数原型

1
2
const int64_t &operator[](const size_t idx) const
int64_t &operator[](const size_t idx)

参数说明

参数

输入/输出

说明

idx

输入

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

返回值说明

  • const int64_t &operator[](const size_t idx) const:步长值,在idx>=kMaxDimNum时,行为未定义。
  • int64_t &operator[](const size_t idx):dim值,在idx>=kMaxDimNum时,行为未定义。

约束说明

调用者需要保证index合法,即idx<kMaxDimNum。

调用示例

1
2
3
4
Stride stride({3, 256, 256});
auto str0 = stride[0]; // 3
auto str5 = stride[5]; // 0
auto invalid_str = stride[Stride::kMaxDimNum]; // 行为未定义