add docstring

jnwnlee · Mar 29, 2024 · 772d2b3 · 772d2b3
1 parent 73f2c0b
commit 772d2b3
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/fadtk/panns/pytorch_utils.py b/fadtk/panns/pytorch_utils.py
@@ -5,6 +5,15 @@
 
 
 def move_data_to_device(x, device):
+    """Move numpy ndarray to device after converting it to torch.Tensor.
+    
+    Args:
+        x: np.ndarray
+        device: torch.device
+        
+    Returns:
+        x: torch.Tensor
+    """
     if "float" in str(x.dtype):
         x = torch.Tensor(x)
     elif "int" in str(x.dtype):