@tensorflow/tfjs-core/dist/kernels/split_shared.js

Hardware-accelerated JavaScript library for machine intelligence

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
function split(x, sizeSplits, axis) {
    var begin = new Array(x.rank).fill(0);
    var size = x.shape.slice();
    return sizeSplits.map(function (s) {
        size[axis] = s;
        var slice = x.slice(begin, size);
        begin[axis] += s;
        return slice;
    });
}
exports.split = split;
//# sourceMappingURL=split_shared.js.map