Layers with recurrent hidden state - rwth-i6/returnn GitHub Wiki
Basically all layers which define get_rec_initial_extra_outputs
.
Most (all) of them only when inside a rec loop, otherwise they behave just as if you would iterate over the time axis.
WindowLayer
:state
[B,W,...]CumsumLayer
:state
which is the output itselfRecLayer
: likeRnnCellLayer
ifunit
is astr
, or like subnet ifunit
is a subnetRnnCellLayer
:state
depending onunit
option, e.g.LSTMStateTuple
, or also often the output itselfSelfAttentionLayer
:k_left
,v_left
KenLmStateLayer
:state
,step
,scores
CumConcatLayer
:state
which is the output itselfSubnetworkLayer
: any from subnetBaseChoiceLayer
,ChoiceLayer
,DecideLayer
, etc:choice_scores
,choice_src_beams
EditDistanceTableLayer
:state
, optionalsource_len
MaskedComputationLayer
: any from subnet,_output
UnmaskLayer
:t
TwoDLSTMLayer
:state
,output
,iteration