RSVP for your your local TensorFlow Everywhere event today!

tf.compat.v2.nn.RNNCellResidualWrapper

View source on GitHub

RNNCell wrapper that ensures cell inputs are added to the outputs.

cell An instance of RNNCell.
residual_fn (Optional) The function to map raw cell inputs and raw cell outputs to the actual cell outputs of the residual network. Defaults to calling nest.map_structure on (lambda i, o: i + o), inputs and outputs.
**kwargs dict of keyword arguments for base layer.

output_size

state_size

Methods

get_initial_state

View source

zero_state

View source