mlts-experiment-data/dist/src/utils/utils.js

Machine learning experiment data downloader.

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
var tslib_1 = require("tslib");
function tuple(t1, t2) {
    return [t1, t2];
}
exports.tuple = tuple;
function sum(arr) {
    return arr.reduce(function (x, y) { return x + y; }, 0);
}
exports.sum = sum;
function product(arr) {
    return arr.reduce(function (x, y) { return x * y; }, 1);
}
exports.product = product;
function getRow(tensor, row) {
    var cols = product(tensor.shape.slice(1));
    var start = row * cols;
    return tensor.data.slice(start, start + cols);
}
exports.getRow = getRow;
function asArray(arr) {
    var out = [];
    for (var i = 0; i < arr.length; ++i)
        out.push(arr[i]);
    return out;
}
exports.asArray = asArray;
function splitTensor(tensor, splitPoint) {
    var cols = product(tensor.shape.slice(1));
    var firstData = tensor.data.slice(0, splitPoint * cols);
    var secondData = tensor.data.slice(splitPoint * cols, tensor.data.length);
    var first = {
        data: firstData,
        type: tensor.type,
        shape: tslib_1.__spread([
            splitPoint
        ], tensor.shape.slice(1)),
    };
    var second = {
        data: secondData,
        type: tensor.type,
        shape: tslib_1.__spread([
            tensor.shape[0] - splitPoint
        ], tensor.shape.slice(1)),
    };
    return tuple(first, second);
}
exports.splitTensor = splitTensor;
//# sourceMappingURL=utils.js.map