print the value of variables after each line of code

Question

I\'m trying to understand how the following PyTorch code works. To know how each function works and what they output & to know the outputted variables value and size, I\'m u