@hoff97/tensor-js/dist/lib/model/functional/bce/cpu.js

PyTorch like deep learning inferrence library

import { positionWiseBinaryOp } from '../../../ops/cpu/basic';
export function bce(x, y) {
    return positionWiseBinaryOp(x, y, (x, y) => {
        if (y === 1) {
            return -Math.log(x);
        }
        else {
            return -Math.log(1 - x);
        }
    }, x.shape);
}
//# sourceMappingURL=cpu.js.map