Options
All
  • Public
  • Public/Protected
  • All
Menu

Helper for a Bidirectional LSTM layer.

Implements processing the input in chunks, which is significantly more efficient in tfjs due to memory management and shader caching.

Hierarchy

  • Lstm

Index

Constructors

Methods

Constructors

constructor

  • new Lstm(inputShape: number[], units?: number): Lstm

Methods

dispose

  • dispose(): void

predict

  • predict(inputs: tf.Tensor3D, chunkSize: number): Tensor<R3>

setWeights

  • setWeights(vars: tf.NamedTensorMap, scope: string, denseName: string): void

Generated using TypeDoc