Update softmax.cairo

gizatechxyz · Oct 22, 2023 · a835f73 · a835f73
1 parent 73e49ab
commit a835f73
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/operators/nn/functional/softmax.cairo b/src/operators/nn/functional/softmax.cairo
@@ -55,7 +55,7 @@ use orion::operators::tensor::{
 /// Cf: NNTrait::softmax docstring
 fn softmaxWide2(z: @Tensor<FP16x16>, axis: usize) -> Tensor<FP16x16W> {
     let exp_tensor: Tensor<FP16x16W> = exp_upcast(*z);
-    let sum = exp_tensor.reduce_sum(axis, true);
+    // let sum = exp_tensor.reduce_sum(axis, true);
     // let softmax = exp_tensor / sum;
-    return sum;
+    return exp_tensor;
 }