refactor(tensor): simplify select_dim signature by replacing generic type constraints with impl AsIndex, update related usage in silero_vad module (#152)

Author: crutcher

crates/bunsen/src/burner/tensor/tensor_op_ext.rs

@@ -34,10 +34,10 @@ where
     fn extract(&mut self) -> Self;
 
     /// Select (and Squeeze) a dimension.
-    fn select_dim<const D2: usize, I: AsIndex>(
+    fn select_dim<const D2: usize>(
         self,
-        dim: usize,
-        index: I,
+        dim: impl AsIndex,
+        index: impl AsIndex,
     ) -> Tensor<B, D2, K>;
 }
 
@@ -59,13 +59,14 @@ where
         z
     }
 
-    fn select_dim<const D2: usize, I: AsIndex>(
+    fn select_dim<const D2: usize>(
         self,
-        dim: usize,
-        index: I,
+        dim: impl AsIndex,
+        index: impl AsIndex,
     ) -> Tensor<B, D2, K> {
+        let dim = dim.expect_dim_index(D);
         let index = index.as_index();
-        self.slice_dim(dim, index..index + 1).squeeze_dim::<D2>(dim)
+        self.slice_dim(dim, index).squeeze_dim::<D2>(dim)
     }
 }
 

crates/bunsen/src/kits/speech/silero_vad/blocks/module.rs

@@ -648,7 +648,7 @@ impl<B: Backend> SileroVad<B> {
             (mut $acc:ident) => {{
                 for step in 0..steps {
                     // [batch, d_hidden]
-                    let features = seq_features.clone().select_dim::<2, _>(0, step);
+                    let features = seq_features.clone().select_dim::<2>(0, step);
 
                     // [batch, d_hidden]
                     (hidden, cell) = self.lstm_step(features, hidden, cell);
@@ -773,7 +773,7 @@ impl<B: Backend> SileroVad<B> {
         let mag = (real_2 + imag_2).sqrt();
 
         // Encode, then take the first (and, for a single chunk, only) frame.
-        let x = self.encoder.forward(mag).select_dim::<2, _>(2, 0);
+        let x = self.encoder.forward(mag).select_dim::<2>(2, 0);
 
         #[cfg(any(test, debug_assertions))]
         crate::contracts::assert_shape_contract_periodically!(