Function customLoss

tfjs-layers/src/losses_test.ts:395–395

(x: Tensor, y: Tensor)

  it(`get custom loss works`, () => {
    const customLoss = (x: Tensor, y: Tensor) => scalar(42.0);
    expect(losses.get(customLoss)).toEqual(customLoss);
  });
});