Tensorflow предлагает хорошую оболочку LSTM.

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

Я хотел бы использовать регуляризацию, скажем, регуляризацию L2. Однако у меня нет прямого доступа к различным весовым матрицам, используемым в ячейке LSTM, поэтому я не могу явно сделать что-то вроде

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

Есть ли способ получить доступ к матрицам или как-то использовать регуляризацию с помощью LSTM?