Function slice

packages/transformers/src/utils/tensor.js:1141–1150

(data, starts, ends, axes, steps)

 * @returns {Promise<Tensor>} Sliced data tensor.
 */
export async function slice(data, starts, ends, axes, steps) {
    const op = await TensorOpRegistry.slice;
    return await op({
        x: data,
        s: arrayToIndexTensor(starts),
        e: arrayToIndexTensor(ends),
        a: arrayToIndexTensor(axes),
        t: arrayToIndexTensor(steps ?? new Array(axes.length).fill(1)),
    });
}

/**
 * Perform mean pooling of the last hidden state followed by a normalization step.