Hi,
I really thank you a lot for your answer. It fixes my problem  But I have to say the manipulation of the dynamic graph is still complicated for me. Do you know of a good (maybe visual) explanation of that topic? Something like a correspondance between the pytorch commands and what is happening to the graph.