Replace `from torch.nn.functional import ...` with `import torch.nn.functional as F`

[kalekundert]

trafficstars

These two expressions are almost identical, but it turns out that the latter is (currently) necessary to visualize models using torchlens. See johnmarktaylor91/torchlens#18 for more information. While this is a pretty minor benefit, it's also a pretty minor change.