Extracting weights from trained GRU model

C++
1 
I am implementing c++ code for GRU.

Trained using:
nn.GRU(input_size=32, hidden_size=32, num_layers=1, dropout=0, batch_first=True)

For that I have extracted weights from model using model.weight_ih_l0 and model.weight_hh_10. They are coming as a matrix with 96x32. How can I extract reset and update gates weights from them to calculate equations? Hope I get a solution.

2

The Variables section of the GRU documentation has the order relating to the equations at the top.

Best regards

Thomas