Function _binaryCrossentropy

tfjs-layers/src/losses_test.ts:315–323

(target: Tensor, output: Tensor)

describeMathCPUAndGPU('binaryCrossentropy', () => {
  function _binaryCrossentropy(target: Tensor, output: Tensor): Tensor {
    const targetComplement = tfc.add(scalar(1), tfc.neg(target));
    const outputComplement = tfc.add(scalar(1), tfc.neg(output));
    return tfc.mean(
        tfc.neg(tfc.add(
            tfc.mul(target, tfc.log(output)),
            tfc.mul(targetComplement, tfc.log(outputComplement)))),
        -1);
  }

  it('from sigmoid', () => {
    const x = tensor2d([[0.3, 0.7], [0.4, 0.6]], [2, 2]);