目录

operator[]

函数功能

获取指定index轴的dim值。

函数原型

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

参数说明

参数 输入/输出 说明
idx 输入 dim的index,调用者需要保证index合法。

返回值

约束说明

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

调用示例

Shape shape0({3, 256, 256});
auto dim0 = shape0[0]; // 0轴的维度大小为3
auto dim1 = shape0[1]; // 1轴的维度大小为256
auto dim2 = shape0[2]; // 2轴的维度大小为256
auto invalid_dim = shape0[kMaxDimNum]; // 行为未定义