tf.compat.v1.nn.rnn_cell.ResidualWrapper
RNNCell wrapper that ensures cell inputs are added to the outputs.
Inherits From: RNNCell
, Layer
, Layer
, Module
tf.compat.v1.nn.rnn_cell.ResidualWrapper( *args, **kwargs )
Args | |
---|---|
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. |
Attributes | |
---|---|
graph | |
output_size | |
scope_name | |
state_size |
Methods
get_initial_state
get_initial_state( inputs=None, batch_size=None, dtype=None )
zero_state
zero_state( batch_size, dtype )
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/compat/v1/nn/rnn_cell/ResidualWrapper