@tensorflow/tfjs-core
Version:
Hardware-accelerated JavaScript library for machine intelligence
11 lines (10 loc) • 417 B
TypeScript
import { Scalar, Variable } from '../tensor';
import { NamedTensorMap } from '../types';
export declare abstract class Optimizer {
minimize(f: () => Scalar, returnCost?: boolean, varList?: Variable[]): Scalar | null;
computeGradients(f: () => Scalar, varList?: Variable[]): {
value: Scalar;
grads: NamedTensorMap;
};
abstract applyGradients(variableGradients: NamedTensorMap): void;
}