On this page
tf.nn.RNNCellResidualWrapper
RNNCell wrapper that ensures cell inputs are added to the outputs.
Inherits From: AbstractRNNCell
, Layer
, Module
tf.nn.RNNCellResidualWrapper(
*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 | |
---|---|
output_size |
|
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/nn/RNNCellResidualWrapper