[DOCS] Fixed KLDiv example (#126857)

elisim · pytorchmergebot · commit 2ac739cc80fe · 2024-05-24T15:39:50.000Z
Small import fix to make the example run Pull Request resolved: #126857 Approved by: https://github.com/albanD
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)
