This is an adaptor for RNNCell classes to be used with FusedRNNCell
.
Inherits From: FusedRNNCell
tf.contrib.rnn.FusedRNNCellAdaptor(
cell, use_dynamic_rnn=False
)
Args |
cell
|
an instance of a subclass of a rnn_cell.RNNCell .
|
use_dynamic_rnn
|
whether to use dynamic (or static) RNN.
|
Methods
__call__
View source
__call__(
inputs, initial_state=None, dtype=None, sequence_length=None, scope=None
)
Run this fused RNN on inputs, starting from the given state.
Args |
inputs
|
3-D tensor with shape [time_len x batch_size x input_size]
or a list of time_len tensors of shape [batch_size x input_size] .
|
initial_state
|
either a tensor with shape [batch_size x state_size]
or a tuple with shapes [batch_size x s] for s in state_size , if the
cell takes tuples. If this is not provided, the cell is expected to
create a zero initial state of type dtype .
|
dtype
|
The data type for the initial state and expected output. Required
if initial_state is not provided or RNN state has a heterogeneous
dtype.
|
sequence_length
|
Specifies the length of each sequence in inputs. An
int32 or int64 vector (tensor) size [batch_size] , values in [0,
time_len) .
Defaults to time_len for each element.
|
scope
|
VariableScope or string for the created subgraph; defaults to
class name.
|
Returns |
A pair containing:
- Output: A
3-D tensor of shape [time_len x batch_size x output_size]
or a list of time_len tensors of shape [batch_size x output_size] ,
to match the type of the inputs .
- Final state: Either a single
2-D tensor, or a tuple of tensors
matching the arity and shapes of initial_state .
|