Update softmax.cairo

gizatechxyz · Oct 22, 2023 · febcae7 · febcae7
1 parent 162f528
commit febcae7
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/operators/nn/functional/softmax.cairo b/src/operators/nn/functional/softmax.cairo
@@ -56,7 +56,9 @@ use debug::PrintTrait;
 /// Cf: NNTrait::softmax docstring
 fn softmaxWide2(z: @Tensor<FP16x16>, axis: usize) -> Tensor<FP16x16> {
     let exp_tensor: Tensor<FP16x16W> = exp_upcast(*z);
-    (exp_tensor.data.len()).print();
+    (*exp_tensor.data.at(0)).print();
+    (*exp_tensor.data.at(1)).print();
+    (*exp_tensor.data.at(2)).print();
 
     // let sum = exp_tensor.reduce_sum(axis, true);
     // (*sum.data.at(0)).print();