@tensorflow/tfjs-core
Version:
Hardware-accelerated JavaScript library for machine intelligence
9 lines (8 loc) • 483 B
TypeScript
import { Scalar, Tensor1D, Tensor2D } from '../tensor';
export declare type LSTMCellFunc = {
(data: Tensor2D, c: Tensor2D, h: Tensor2D): [Tensor2D, Tensor2D];
};
export declare class LSTMOps {
static multiRNNCell(lstmCells: LSTMCellFunc[], data: Tensor2D, c: Tensor2D[], h: Tensor2D[]): [Tensor2D[], Tensor2D[]];
static basicLSTMCell(forgetBias: Scalar, lstmKernel: Tensor2D, lstmBias: Tensor1D, data: Tensor2D, c: Tensor2D, h: Tensor2D): [Tensor2D, Tensor2D];
}