Cells
LuxRecurrentLayers.AntisymmetricRNNCell
— TypeAntisymmetricRNNCell(in_dims => out_dims, [activation];
init_weight = nothing, init_recurrent_weight = nothing,
bias = true, epsilon=1.0)
Arguments
input_size => hidden_size
: input and inner dimension of the layeractivation
: activation function. Default istanh
Keyword arguments
init_weight
: initializer for the input to hidden weightsinit_recurrent_weight
: initializer for the hidden to hidden weightsbias
: include a bias or not. Default istrue
epsilon
: step size. Default is 1.0.gamma
: strength of diffusion. Default is 0.0
Equations
\[h_t = h_{t-1} + \epsilon \tanh \left( (W_h - W_h^T - \gamma I) h_{t-1} + V_h x_t + b_h \right),\]