brain.js
Version:
Neural networks in JavaScript
47 lines • 2.73 kB
TypeScript
import { IConstantsThis, IKernelFunctionThis, IKernelRunShortcut, KernelOutput } from 'gpu.js';
import { Filter, IFilterSettings } from './filter';
import { ILayer } from './base-layer';
export interface IPredictConstants extends IConstantsThis {
inputWidth: number;
inputHeight: number;
}
export declare function predict(this: IKernelFunctionThis<IPredictConstants>, inputs: number[][], filters: number[][], biases: number[]): number;
export declare function predict3D(this: IKernelFunctionThis<IPredictConstants>, inputs: number[][][], filters: number[][], biases: number[]): number;
export interface ICompareInputDeltasConstants extends IConstantsThis {
filterCount: number;
}
export declare function compareInputDeltas(this: IKernelFunctionThis<ICompareInputDeltasConstants>, inputDeltas: number[][], deltas: number[][], filters: number[][]): number;
export declare function compareInputDeltas3D(this: IKernelFunctionThis<ICompareInputDeltasConstants>, inputDeltas: number[][][], deltas: number[][], filters: number[][]): number;
export declare function compareBiases(this: IKernelFunctionThis, biases: number[], deltas: number[][]): number;
export interface ICompareFiltersDeltas extends IConstantsThis {
deltaX: number;
deltaY: number;
inputWidth: number;
inputHeight: number;
}
export declare function compareFilterDeltas(this: IKernelFunctionThis<ICompareFiltersDeltas>, filterDeltas: number[][], inputWeights: number[][], deltas: number[][]): number;
export declare function compareFilterDeltas3D(this: IKernelFunctionThis<ICompareFiltersDeltas>, filterDeltas: number[][], inputWeights: number[][][], deltas: number[][]): number;
export interface IFullyConnectedDefaultSettings extends Partial<IFilterSettings> {
bias?: number;
biases?: KernelOutput;
biasDeltas?: KernelOutput;
}
export declare const defaults: IFullyConnectedDefaultSettings;
export declare class FullyConnected extends Filter {
get bias(): number;
get biases(): KernelOutput;
set biases(biases: KernelOutput);
get biasDeltas(): KernelOutput;
set biasDeltas(biasDeltas: KernelOutput);
settings: Partial<IFullyConnectedDefaultSettings>;
compareFilterDeltasKernel: IKernelRunShortcut | null;
compareInputDeltasKernel: IKernelRunShortcut | null;
compareBiasesKernel: IKernelRunShortcut | null;
constructor(settings: Partial<IFullyConnectedDefaultSettings>, inputLayer: ILayer);
validate(): void;
setupKernels(): void;
predict(): void;
compare(): void;
}
export declare function fullyConnected(settings: IFullyConnectedDefaultSettings, inputLayer: ILayer): FullyConnected;
//# sourceMappingURL=fully-connected.d.ts.map