diff options
Diffstat (limited to 'arm_compute/core')
-rw-r--r-- | arm_compute/core/Dimensions.h | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/arm_compute/core/Dimensions.h b/arm_compute/core/Dimensions.h index 3d9a3fa7ff..912b9d57d7 100644 --- a/arm_compute/core/Dimensions.h +++ b/arm_compute/core/Dimensions.h @@ -141,6 +141,17 @@ public: std::fill(_id.begin() + _num_dimensions, _id.end(), 0); } + /** Collapse dimensions starting from a given point + * + * @param[in] start Starting point of collapsing dimensions + */ + void collapse_from(size_t start) + { + ARM_COMPUTE_ERROR_ON(start > num_dimensions()); + + collapse(num_dimensions() - start, start); + } + /** Returns a read/write iterator that points to the first element in the dimension array. */ typename std::array<T, num_max_dimensions>::iterator begin() { |