GetDim

函数功能

获取对应idx轴的dim值。

函数原型

int64_t GetDim(const size_t idx) const

参数说明

返回值说明

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