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:statewhich is the output itselfRecLayer: likeRnnCellLayerifunitis astr, or like subnet ifunitis a subnetRnnCellLayer:statedepending onunitoption, e.g.LSTMStateTuple, or also often the output itselfSelfAttentionLayer:k_left,v_leftKenLmStateLayer:state,step,scoresCumConcatLayer:statewhich is the output itselfSubnetworkLayer: any from subnetBaseChoiceLayer,ChoiceLayer,DecideLayer, etc:choice_scores,choice_src_beamsEditDistanceTableLayer:state, optionalsource_lenMaskedComputationLayer: any from subnet,_outputUnmaskLayer:tTwoDLSTMLayer:state,output,iteration