docs

RichieHakim · Apr 8, 2024 · c052105 · c052105
1 parent 743a35a
commit c052105
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/bnpm/torch_helpers.py b/bnpm/torch_helpers.py
@@ -893,6 +893,7 @@ def slice_along_dim(
 ) -> torch.Tensor:
     """
     Slices a tensor along a specified dimension.
+    RH 2022
 
     Args:
         X (torch.Tensor): 
@@ -906,8 +907,6 @@ def slice_along_dim(
         (torch.Tensor): 
             sliced_tensor (torch.Tensor):
                 Sliced tensor.
-
-    RH 2022
     """
     slices = [slice(None)] * X.ndim
     slices[dim] = idx