refactor: import

Author: kozistr

pytorch_optimizer/optimizer/utils.py

@@ -7,7 +7,7 @@
 import torch
 from torch import nn
 from torch.distributed import all_reduce
-from torch.nn import functional as f
+from torch.nn.functional import cosine_similarity
 from torch.nn.modules.batchnorm import _BatchNorm
 from torch.nn.utils import clip_grad_norm_
 
@@ -119,7 +119,7 @@ def cosine_similarity_by_view(
     """
     x = view_func(x)
     y = view_func(y)
-    return f.cosine_similarity(x, y, dim=1, eps=eps).abs_()
+    return cosine_similarity(x, y, dim=1, eps=eps).abs_()
 
 
 def clip_grad_norm(