Actually, I don’t see why I should specify retain_graph = True because I’m only using chunks of this tensor which are only used once. And, as I said, the next state value has no history, I use it as a label.