What are the differences between torch.nn and torch.nn.functional and when are they used?
Have a look at this post for some explanations.
