-
Notifications
You must be signed in to change notification settings - Fork 133
Layers with recurrent hidden state
Albert Zeyer edited this page Oct 27, 2021
·
4 revisions
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 itself -
RecLayer: likeRnnCellLayerifunitis astr, or like subnet ifunitis a subnet -
RnnCellLayer:statedepending onunitoption, e.g.LSTMStateTuple, or also often the output itself -
SelfAttentionLayer:k_left,v_left -
KenLmStateLayer:state,step,scores -
CumConcatLayer:statewhich is the output itself -
SubnetworkLayer: any from subnet -
BaseChoiceLayer,ChoiceLayer,DecideLayer, etc:choice_scores,choice_src_beams -
EditDistanceTableLayer:state, optionalsource_len -
MaskedComputationLayer: any from subnet,_output -
UnmaskLayer:t -
TwoDLSTMLayer:state,output,iteration