grain ~master (2021-05-15T05:53:16Z)
Dub
Repo
RNNLM
ptb
Undocumented in source.
struct
RNNLM (
alias
Storage
) {
Embedding
!(
float
,
Storage
)
embed
;
LSTM
!(
float
,
Storage
)
lstm1
;
LSTM
!(
float
,
Storage
)
lstm2
;
Linear
!(
float
,
Storage
)
linear
;
static
ref
init
(C chain, Args args);
this
(int vocabSize, int embedSize, int hiddenSize);
auto
opCall
(Slice!(int*, 2) xslice, O optimizer);
}
Constructors
this
this
(int vocabSize, int embedSize, int hiddenSize)
Undocumented in source.
Members
Functions
opCall
auto
opCall
(Slice!(int*, 2) xslice, O optimizer)
Undocumented in source. Be warned that the author may not have intended to support it.
Static functions
init
ref
init
(C chain, Args args)
Undocumented in source. Be warned that the author may not have intended to support it.
Variables
embed
Embedding
!(
float
,
Storage
)
embed
;
Undocumented in source.
linear
Linear
!(
float
,
Storage
)
linear
;
Undocumented in source.
lstm1
LSTM
!(
float
,
Storage
)
lstm1
;
lstm2
LSTM
!(
float
,
Storage
)
lstm2
;
Undocumented in source.
Meta
Source
See Implementation
ptb
functions
main
test
uniform_
structs
LSTM
RNNLM