Function reshape

packages/transformers/src/utils/tensor.js:974–1003

* Reshapes a 1-dimensional array into an n-dimensional array, according to the provided dimensions. * * @example * reshape([10 ], [1 ]); // Type: number[] Value: [10] * reshape([1, 2, 3, 4 ], [2, 2 ]); // Type: number[][] Value: [[1, 2], [3, 4]]

(data, dimensions)

 * @returns {NestArray<T, DIM["length"]>} The reshaped array.
 */
function reshape(data, dimensions) {
    const totalElements = data.length;
    const dimensionSize = dimensions.reduce((a, b) => a * b);

    if (totalElements !== dimensionSize) {
        throw Error(`cannot reshape array of size ${totalElements} into shape (${dimensions})`);
    }

    /** @type {any} */
    let reshapedArray = data;

    for (let i = dimensions.length - 1; i >= 0; i--) {
        reshapedArray = reshapedArray.reduce(
            (acc, val) => {
                let lastArray = acc[acc.length - 1];

                if (lastArray.length < dimensions[i]) {
                    lastArray.push(val);
                } else {
                    acc.push([val]);
                }

                return acc;
            },
            [[]],
        );
    }

    return reshapedArray[0];
}

/**
 * Permutes a tensor according to the provided axes.