refactor: remove old softmax function

besaleli · besaleli · commit b7509f3ab3c3 · 2025-11-06T19:04:09.000-05:00
diff --git a/encoderfile/src/inference/utils.rs b/encoderfile/src/inference/utils.rs
@@ -55,14 +55,6 @@ pub fn l2_normalize(mut x: Array2<f32>, axis: Axis) -> Array2<f32> {
     x
 }
 
-pub fn softmax(x: &Array2<f32>, axis: Axis) -> Array2<f32> {
-    let max_per_axis = x.map_axis(axis, |row| row.fold(f32::NEG_INFINITY, |a, &b| a.max(b)));
-    let expx = x - &max_per_axis.insert_axis(axis);
-    let expx = expx.mapv(f32::exp);
-    let sum = expx.sum_axis(axis).insert_axis(axis);
-    &expx / &sum
-}
-
 pub fn requires_token_type_ids<'a>(session: &MutexGuard<'a, Session>) -> bool {
     session
         .inputs

Original file line number	Diff line number	Diff line change
`@@ -55,14 +55,6 @@ pub fn l2_normalize(mut x: Array2<f32>, axis: Axis) -> Array2<f32> {`
`55`	`55`	`x`
`56`	`56`	`}`
`57`	`57`
`58`		`-pub fn softmax(x: &Array2<f32>, axis: Axis) -> Array2<f32> {`
`59`		`- let max_per_axis = x.map_axis(axis, \|row\| row.fold(f32::NEG_INFINITY, \|a, &b\| a.max(b)));`
`60`		`- let expx = x - &max_per_axis.insert_axis(axis);`
`61`		`- let expx = expx.mapv(f32::exp);`
`62`		`- let sum = expx.sum_axis(axis).insert_axis(axis);`
`63`		`- &expx / &sum`
`64`		`-}`
`65`		`-`
`66`	`58`	`pub fn requires_token_type_ids<'a>(session: &MutexGuard<'a, Session>) -> bool {`
`67`	`59`	`session`
`68`	`60`	`.inputs`