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 22), but it will be too complicated for me.
Is there any way to access internal variables?
Thank you in advance.