rnn.graph.unroll
¶
Description¶
Unroll representation of RNN running on non CUDA device
Usage¶
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 |
---|---|
|
int, number of stacked layers |
|
int, number of time steps to unroll |
|
int, number of levels in the data - only used for embedding |
|
int, default = NULL - no embedding. Dimension of the embedding vectors |
|
int, size of the state in each RNN layer |
|
int, number of output variables in the decoding layer |
|
|
|
Either seq-to-one or one-to-one |
|
Type of RNN cell: either gru or lstm |