LSTM¶

Applies a multi-layer long short-term memory (LSTM) RNN.

Abstract Signature:

LSTM(input_size: int, hidden_size: int, num_layers: int = 1)

API: torch.nn.LSTM

Strategy: Direct Mapping

API: keras.layers.LSTM

Strategy: Direct Mapping

API: tf.keras.layers.LSTM

Strategy: Direct Mapping

API: mlx.nn.layers.recurrent.LSTM

Strategy: Direct Mapping

API: flax.nnx.LSTM

Strategy: Direct Mapping

API: praxis.layers.LSTM

Strategy: Direct Mapping