O Tensorflow oferece um bom wrapper LSTM.

rnn_cell.BasicLSTM(num_units, forget_bias=1.0, input_size=None,
           state_is_tuple=False, activation=tanh)

Eu gostaria de usar a regularização, digamos, a regularização L2. No entanto, não tenho acesso direto às diferentes matrizes de peso usadas na célula LSTM, portanto não posso fazer algo explicitamente como

loss = something + beta * tf.reduce_sum(tf.nn.l2_loss(weights))

Existe uma maneira de acessar as matrizes ou usar a regularização de alguma forma com o LSTM?