I’m not sure I understand the issue correctly, so please add more information in case I misunderstand it indeed.



 Stefano_Barone:

I would like to create a dataloader with only the features data and y_vector.


Wouldn’t removing self.z_np_vector work in this case? It seems, this particular tensor is also not needed in the model, so I’m unsure why it’s still there.
However, this description seems to point towards a usage of this tensor:



 Stefano_Barone:

I want the nth element of the test_dataset if I don’t include z_vector I can’t understand which element of z_vector I have to take.


Where is this tensor used, if not in the model?