* @brief Returns the dimension of the index-th axis (or the negative index-th * axis from the end, if index is negative). * * @param index the axis index, which may be negative as it will be * "canonicalized" using CanonicalAxisIndex. * Dies on out of range index. */
| 69 | * Dies on out of range index. |
| 70 | */ |
| 71 | inline int shape(int index) const { |
| 72 | return shape_[CanonicalAxisIndex(index)]; |
| 73 | } |
| 74 | inline int num_axes() const { return shape_.size(); } |
| 75 | inline int count() const { return count_; } |
| 76 |
no outgoing calls