recurrent-js-gpu
Version:
GPU-accelerated Deep Recurrent Neural Networks and LSTMs in Typescript. Ported, object-oriented and refactored version of Andrej Karpathy's recurrent-js (https://github.com/karpathy/recurrentjs)
12 lines (11 loc) • 485 B
TypeScript
import { Mat } from './Mat';
import { Graph } from './Graph';
import { NNModel } from './NNModel';
import { PreviousOutput } from './utils/PreviousOutput';
export declare class LSTM extends NNModel {
inputSize: number;
hiddenSizes: Array<number>;
outputSize: number;
constructor(inputSize: number, hiddenSizes: Array<number>, outputSize: number, needsBackProp?: boolean);
forward(observations: Mat, previousOutput: PreviousOutput, graph?: Graph): PreviousOutput;
}