[DOCS] Fixed KLDiv example (#126857)

Small import fix to make the example run Pull Request resolved: #126857 Approved by: https://github.com/albanD
pytorch · May 24, 2024 · 2ac739c · 2ac739c
1 parent 4105f91
commit 2ac739c
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/torch/nn/modules/loss.py b/torch/nn/modules/loss.py
@@ -447,8 +447,6 @@ class KLDivLoss(_Loss):
           same shape as the input.
 
     Examples::
-
-        >>> import torch.nn.functional as F
         >>> kl_loss = nn.KLDivLoss(reduction="batchmean")
         >>> # input should be a distribution in the log space
         >>> input = F.log_softmax(torch.randn(3, 5, requires_grad=True), dim=1)