How to make ctags work for pytorch

Hi,

could anyone advice me, how make ctags work for pytorch code? It works fine for fastai, but sometimes I am curious about pytorch implementions as well.