ml-basic/dist/types/layers/lstm.d.ts

Lightweight, zero dependency, machine learning library

import { Sigmoid, TanH } from "../lib/functions";
import Matrix from "../lib/matrix";
import LoopLayer, { LoopParams } from "./loop";
export default class LSTMLayer extends LoopLayer<'o' | 'u' | 'c' | 'memory' | 'state' | 'input' | 'output'> {
    name: string;
    yWeights: Matrix;
    yBias: Matrix;
    fWeights: Matrix;
    fBias: Matrix;
    oWeights: Matrix;
    oBias: Matrix;
    uWeights: Matrix;
    uBias: Matrix;
    cWeights: Matrix;
    cBias: Matrix;
    memory: Matrix;
    sigmoid: Sigmoid;
    tanh: TanH;
    constructor(args: LoopParams);
    clear(): void;
    forward(input: Matrix, output: boolean): Matrix | undefined;
    backward(loss: Matrix): Matrix;
}