flax.linen.RNNCellBase#

class flax.linen.RNNCellBase(parent=<flax.linen.module._Sentinel object>, name=None)[source]#

RNN cell base class.

__call__(**kwargs)#

Call self as a function.

Methods

initialize_carry(rng, input_shape)

Initialize the RNN cell carry.