UNPKG

@tensorflow/tfjs-core

Version:

Hardware-accelerated JavaScript library for machine intelligence

12 lines (11 loc) 498 B
import { Serializable } from '../serialization'; import { Scalar, Variable } from '../tensor'; import { NamedTensorMap } from '../types'; export declare abstract class Optimizer extends Serializable { minimize(f: () => Scalar, returnCost?: boolean, varList?: Variable[]): Scalar | null; computeGradients(f: () => Scalar, varList?: Variable[]): { value: Scalar; grads: NamedTensorMap; }; abstract applyGradients(variableGradients: NamedTensorMap): void; }