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