But in the second example, as you see I am printing the input before sending it to the forward method, don’t I? It’s the same as example in the PyTorch docs.
And would you please tell me how can I add print statement to the forward method, directly?