GetDim

函数功能

获取对应idx轴的dim值。

函数原型

int64_t GetDim(const size_t idx) const

参数说明

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

返回值说明

dim值，在idx>=kMaxDimNum时，返回`kInvalidDimValue`。

约束说明

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

调用示例

Shape shape0({3, 256, 256});
auto dim0 = shape0.GetDim(0); // 3
auto invalid_dim = shape0.GetDim(kMaxDimNum); // kInvalidDimValue