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

PyTorch like deep learning inferrence library

import { Tensor } from '../../../library';
import { DType } from '../../../types';
/**
 * Calculates the binary cross entropy loss, given probabilities x
 * and ground truth y. Returns a tensor of the same shape as
 * x. To use for a loss, you have to sum over the result:
 * ```typescript
 * const loss = bce(x,y).sum();
 * ```
 *
 * @param x Probabilities in [0,1]
 * @param y Ground truth labels of the same shape as x.
 */
export declare function bce<DTpe extends DType>(x: Tensor<DTpe>, y: Tensor<DTpe>): Tensor<DTpe>;