blob: df32c64736771af3f2ea880f9a39c56c07c49b85 [file] [log] [blame]
``rnn.graph.unroll``
========================================
Description
----------------------
Unroll representation of RNN running on non CUDA device
Usage
----------
.. code:: r
rnn.graph.unroll(
num_rnn_layer,
seq_len,
input_size = NULL,
num_embed = NULL,
num_hidden,
num_decode,
dropout = 0,
ignore_label = -1,
loss_output = NULL,
init.state = NULL,
config,
cell_type = "lstm",
masking = F,
output_last_state = F,
prefix = "",
data_name = "data",
label_name = "label"
)
Arguments
------------------
+----------------------------------------+------------------------------------------------------------+
| Argument | Description |
+========================================+============================================================+
| ``num_rnn_layer`` | int, number of stacked layers |
+----------------------------------------+------------------------------------------------------------+
| ``seq_len`` | int, number of time steps to unroll |
+----------------------------------------+------------------------------------------------------------+
| ``input_size`` | int, number of levels in the data - only used for |
| | embedding |
+----------------------------------------+------------------------------------------------------------+
| ``num_embed`` | int, default = NULL - no embedding. Dimension of the |
| | embedding |
| | vectors |
+----------------------------------------+------------------------------------------------------------+
| ``num_hidden`` | int, size of the state in each RNN layer |
+----------------------------------------+------------------------------------------------------------+
| ``num_decode`` | int, number of output variables in the decoding layer |
+----------------------------------------+------------------------------------------------------------+
| ``dropout`` | |
+----------------------------------------+------------------------------------------------------------+
| ``config`` | Either seq-to-one or one-to-one |
+----------------------------------------+------------------------------------------------------------+
| ``cell_type`` | Type of RNN cell: either gru or lstm |
+----------------------------------------+------------------------------------------------------------+