Method collapse

arm_compute/core/TensorShape.h:138–144

Collapse the first n dimensions. * * @param[in] n Number of dimensions to collapse into @p first * @param[in] first Dimensions into which the following @p n are collapsed. */

     * @param[in] first Dimensions into which the following @p n are collapsed.
     */
    void collapse(size_t n, size_t first = 0)
    {
        Dimensions::collapse(n, first);

        // Make sure all empty dimensions are filled with 1
        std::fill(_id.begin() + _num_dimensions, _id.end(), 1);
    }
    /** Shifts right the tensor shape increasing its dimensions
     *
     * @param[in] step Rotation step