Method remove_dimension

arm_compute/core/TensorShape.h:113–131

Accessor to remove the dimension n from the tensor shape. * * @note The upper dimensions of the tensor shape will be shifted down by 1 * * @param[in] n Dimension to remove * @param[in] apply_dim_correction (Optional) Flag to state whether apply dimension correction (removing trailing dimensions with size of 1) after removing a dimension. */

     * @param[in] apply_dim_correction (Optional) Flag to state whether apply dimension correction (removing trailing dimensions with size of 1) after removing a dimension.
     */
    void remove_dimension(size_t n, bool apply_dim_correction = true)
    {
        ARM_COMPUTE_ERROR_ON(_num_dimensions < 1);
        ARM_COMPUTE_ERROR_ON(n >= _num_dimensions);

        std::copy(_id.begin() + n + 1, _id.end(), _id.begin() + n);

        // Reduce number of dimensions
        _num_dimensions--;

        // Make sure all empty dimensions are filled with 1
        std::fill(_id.begin() + _num_dimensions, _id.end(), 1);

        // Correct number dimensions to ignore trailing dimensions of size 1
        if (apply_dim_correction)
        {
            apply_dimension_correction();
        }
    }

    /** Collapse the first n dimensions.
     *