LSTM ==== Computes an one-layer LSTM. This operator is usually supported via some custom implementation such as CuDNN. Notations: * `X` - input tensor * `i` - input gate * `o` - output gate * `f` - forget gate * `c` - cell gate * `t` - time step (t-1 means previous time step) * `W[iofc]` - W parameter weight ... **Abstract Signature:** ``LSTM(X: Tensor, W: Tensor, R: Tensor, B: Tensor, sequence_lens, initial_h: Tensor, initial_c: Tensor, P: Tensor, activation_alpha: List[float], activation_beta: List[float], activations: List[str], clip: float, direction: str, hidden_size: int, input_forget: int, layout: int)`` .. raw:: html

PyTorch

API: torch.nn.modules.rnn.LSTM

Strategy: Direct Mapping

Official Docs ↗

Keras

API: keras.layers.LSTM

Strategy: Direct Mapping

Official Docs ↗

TensorFlow

API: keras.layers.LSTM

Strategy: Direct Mapping

Official Docs ↗

Apple MLX

API: mlx.nn.layers.recurrent.LSTM

Strategy: Direct Mapping

Official Docs ↗