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)

10 lines (9 loc) 437 B
import { Mat } from './Mat'; import { Graph } from './Graph'; import { NNModel } from './NNModel'; import { PreviousOutput } from './utils/PreviousOutput'; export declare class RNN extends NNModel { hiddenSizes: Array<number>; constructor(inputSize: number, hiddenSizes: Array<number>, outputSize: number, needsBackProp?: boolean); forward(observations: Mat, previousOutput: PreviousOutput, graph?: Graph): PreviousOutput; }