I want to get the gate values for the cell state, the reset gate, the update gate of a trained GRU network. According to this issue I\'m able to get weights of kernel, recurrent