I have a AutoEncoder Model in which I return both the encoding and the reconstructed output. This is the forward method that I use:
def forward(self, x):
for module in self.encoder:
x = module(x)
encoding = x*1 # added the multiply just for testing
for module in self.decoder:
x = module(x)
return x, encoding
In line 4, is the simple assign I am doing to copy wrong?
When I check the output, they both refer to different grad_fn
. Am I missing something here?