flax.linen.RNNCellBase

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

RNN cell base class.

__call__(**kwargs)

Call self as a function.

Methods

initialize_carry(rng, input_shape)

Initialize the RNN cell carry.