UNPKG

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
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; }