SimpleCellΒΆ

Simple RNN cell.

Abstract Signature:

SimpleCell(in_features: int, hidden_features: int)

PyTorch

API: torch.nn.RNNCell
Strategy: Direct Mapping

Keras

API: keras.layers.SimpleRNNCell
Strategy: Direct Mapping

TensorFlow

API: tf.keras.layers.SimpleRNNCell
Strategy: Direct Mapping

Flax NNX

API: flax.nnx.SimpleCell
Strategy: Direct Mapping