I’m still unsure, how to interpret this use case and what exactly is not working.
In your previous post you’ve mentioned, that Out.backward() is not working. Could you explain, if you are seeing an error or if this method would break your use case somehow?