@hoff97/tensor-js/dist/lib/ops/gpu/pool/reduceMeanSquare.js

PyTorch like deep learning inferrence library

import { PoolOperation } from './pool';
export class ReduceMeanSquareOperation extends PoolOperation {
    update(a, b) {
        return `(${a}*${a}) + ${b}`;
    }
    post(res) {
        return `${res} = ${res}/float(sumSize);`;
    }
    init(res) {
        return `${res}*${res}`;
    }
}
//# sourceMappingURL=reduceMeanSquare.js.map