Update softmax.cairo

gizatechxyz · Oct 22, 2023 · 162f528 · 162f528
1 parent 803fbce
commit 162f528
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/operators/nn/functional/softmax.cairo b/src/operators/nn/functional/softmax.cairo
@@ -56,7 +56,7 @@ 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.at(0)).print();
+    (exp_tensor.data.len()).print();
 
     // let sum = exp_tensor.reduce_sum(axis, true);
     // (*sum.data.at(0)).print();