From c052105b9c3eabf17fd45dbefce8d63474a55c4b Mon Sep 17 00:00:00 2001
From: RichieHakim <RichHakim@gmail.com>
Date: Mon, 8 Apr 2024 15:08:09 -0400
Subject: [PATCH] docs

---
 bnpm/torch_helpers.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/bnpm/torch_helpers.py b/bnpm/torch_helpers.py
index 21bce82..685285d 100644
--- 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