operator[]

Description

Obtains the dimension value of a specified index axis.

Prototype

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

int64_t &operator[](const size_t idx)

Parameters

Parameter

Input/Output

Description

idx

Input

Index of dim. You must ensure that the index is valid.

Returns

  • const int64_t &operator[](const size_t idx) const: dimension value. When idx is greater than or equal to kMaxDimNum, the behavior is not defined.
  • int64_t &operator[](const size_t idx): dimension value. When idx is greater than or equal to kMaxDimNum, the behavior is not defined.

Restrictions

You must ensure that the index is valid, that is, idx must be less than kMaxDimNum.

Example

Shape shape0({3, 256, 256});
auto dim0 = shape0[0]; // 3
auto dim5 = shape0[5]; // 0
auto invalid_dim = shape0[kMaxDimNum]; // The behavior is not defined.