Create a tensor in global scope whose shape is same a out1. Keep modifying out1, using inplace operations that operate on Model. This way you won’t need to change your forward function’s io.
I assume out1 is not part of the graph…