tf.contrib.rnn.FusedRNNCellAdaptor
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 .
|
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.
Last updated 2020-10-01 UTC.
[null,null,["Last updated 2020-10-01 UTC."],[],[]]