Function stack

packages/transformers/src/utils/tensor.js:1371–1378

(tensors, dim = 0)

 * @returns {Tensor} The stacked tensor.
 */
export function stack(tensors, dim = 0) {
    // TODO do validation of shapes
    // NOTE: stack expects each tensor to be equal size
    return cat(
        tensors.map((t) => t.unsqueeze(dim)),
        dim,
    );
}

/**
 * @param {(previousValue: any, currentValue: any, currentIndex?: number, resultIndex?: number) => any} callbackfn