Какой самый простой способ преобразовать модуль PyTorch (например, RNN, LSTM) в Python график networkx?