tf.nn.RNNCellResidualWrapper

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

Inherits From: AbstractRNNCell, Layer, Module

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