@hoff97/tensor-js/dist/lib/ops/cpu/reduceMeanSquare.js

PyTorch like deep learning inferrence library

import { pool } from './pool';
export function reduceMeanSquare(a, axes, keepDims) {
    let poolSize = 1;
    for (let i = 0; i < axes.length; i++) {
        poolSize *= a.shape[axes[i]];
    }
    return pool(a, axes, (a, b) => {
        return a * a + (b !== undefined ? b : 0);
    }, keepDims, (a) => a / poolSize);
}
//# sourceMappingURL=reduceMeanSquare.js.map