Hi,
I’m not sure what the question is here.
As a note, you should never have to use .data. It should either be replaced by .detach() or with torch.no_grad(): (there are other posts in this forum that discuss this).