Merge pull request #592 from robertknight/fix-arg-min

Fix ArgMin op
robertknight · Feb 10, 2025 · ad555f2 · ad555f2
2 parents 83f0003 + a4d9768
commit ad555f2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/ops/reduce.rs b/src/ops/reduce.rs
@@ -168,7 +168,7 @@ impl Operator for ArgMin {
 
     fn run(&self, pool: &TensorPool, inputs: InputList) -> Result<OutputList, OpError> {
         let input = inputs.require(0)?;
-        dispatch_single_axis_reduce_op!(pool, input, arg_max, self.axis, self.keep_dims)
+        dispatch_single_axis_reduce_op!(pool, input, arg_min, self.axis, self.keep_dims)
     }
 }