How to access internal variables in cudnn GRU


(Minju Jung) #1

I want to access internal variables of “cudnn.CUDNN_GRU”, especially for the candidate hidden state h_tild.
I might access internal variables by changing the descriptor in “lib.cudnnRNNForwardTraining” (https://github.com/pytorch/pytorch/blob/d0cabbde74014a4695d07ca464a21ce7d4a148b8/torch/backends/cudnn/rnn.py#L293), but it will be too complicated for me.

Is there any way to access internal variables?

Thank you in advance.