move imports

torch_sparse/__init__.py

+import torch
+
+torch.ops.load_library('torch_sparse/convert_cpu.so')
+torch.ops.load_library('torch_sparse/diag_cpu.so')
+torch.ops.load_library('torch_sparse/spmm_cpu.so')
+
+try:
+    torch.ops.load_library('torch_sparse/convert_cuda.so')
+    torch.ops.load_library('torch_sparse/diag_cuda.so')
+    torch.ops.load_library('torch_sparse/spmm_cuda.so')
+    torch.ops.load_library('torch_sparse/spspmm_cuda.so')
+except OSError as e:
+    if torch.cuda.is_available():
+        raise e
+
 from .convert import to_torch_sparse, from_torch_sparse, to_scipy, from_scipy
 from .coalesce import coalesce
 from .transpose import transpose

torch_sparse/utils.py

 from typing import Any
 
-import torch
-
 try:
     from typing_extensions import Final  # noqa
 except ImportError:
     from torch.jit import Final  # noqa
 
-torch.ops.load_library('torch_sparse/convert_cpu.so')
-torch.ops.load_library('torch_sparse/diag_cpu.so')
-torch.ops.load_library('torch_sparse/spmm_cpu.so')
-
-try:
-    torch.ops.load_library('torch_sparse/convert_cuda.so')
-    torch.ops.load_library('torch_sparse/diag_cuda.so')
-    torch.ops.load_library('torch_sparse/spmm_cuda.so')
-    torch.ops.load_library('torch_sparse/spspmm_cuda.so')
-except OSError as e:
-    if torch.cuda.is_available():
-        raise e
-
 
 def is_scalar(other: Any) -> bool:
     return isinstance(other, int) or isinstance(other, float)