Function klElement

tfjs-layers/src/losses_test.ts:335–339

(actual: number, predicted: number)

describeMathCPUAndWebGL2('kullbackLeiblerDivergence', () => {
  function klElement(actual: number, predicted: number): number {
    actual = Math.max(actual, epsilon());
    predicted = Math.max(predicted, epsilon());
    return actual * Math.log(actual / predicted);
  }

  it('2D', () => {
    const yTrue = tensor2d([[1, 0], [1, 0]], [2, 2]);