@tensorflow/tfjs-core
Version:
Hardware-accelerated JavaScript library for machine intelligence
12 lines (11 loc) • 437 B
TypeScript
import { Tensor } from '../tensor';
export declare enum Reduction {
NONE = 0,
MEAN = 1,
SUM = 2,
SUM_BY_NONZERO_WEIGHTS = 3,
}
export declare class LossOps {
static computeWeightedLoss<T extends Tensor, O extends Tensor>(losses: T, weights?: Tensor, reduction?: Reduction): O;
static absoluteDifference<T extends Tensor, O extends Tensor>(labels: T, predictions: T, weights?: Tensor, reduction?: Reduction): O;
}