revert to fb6f4a

Author: raphaelDkhn

.DS_Store

@@ -1,7 +1,7 @@
 [package]
 name = "orion"
-version = "0.2.5"
-cairo-version = "2.6.3"
+version = "0.2.4"
+cairo-version = "2.5.3"
 edition = "2023_10"
 description = "ONNX Runtime in Cairo for verifiable ML inference using STARK"
 homepage = "https://github.com/gizatechxyz/orion"

Scarb.toml

@@ -55,8 +55,7 @@ use orion::operators::tensor::TensorTrait;
 | [`tensor.min_in_tensor`](tensor.min\_in\_tensor.md) | Returns the minimum value in the tensor. |
 | [`tensor.min`](tensor.min.md) | Returns the minimum value in the tensor. |
 | [`tensor.max`](tensor.max.md) | Returns the maximum value in the tensor. |
-| [`tensor.reduce_sum`](tensor.reduce\_sum.md) | Computes the sum of the input tensor's elements along the provided axes. |
-| [`tensor.reduce_sum_single_axis`](tensor.reduce\_sum\_single\_axis.md) | Reduces a tensor by summing its elements along a specified axis. |
+| [`tensor.reduce_sum`](tensor.reduce\_sum.md) | Reduces a tensor by summing its elements along a specified axis. |
 | [`tensor.reduce_prod`](tensor.reduce\_prod.md) | Reduces a tensor to its products along specified axis. |
 | [`tensor.argmax`](tensor.argmax.md) | Returns the index of the maximum value along the specified axis. |
 | [`tensor.argmin`](tensor.argmin.md) | Returns the index of the minimum value along the specified axis. |

docs/framework/operators/tensor/README.md

@@ -1,7 +1,7 @@
 #tensor.and
 
 ```rust
-    fn and(self: @Tensor<bool>, other: @Tensor<bool>) -> Tensor<i32>;
+    fn and(self: @Tensor<bool>, other: @Tensor<bool>) -> Tensor<bool>;
 ```
 
 Computes the logical AND of two tensors element-wise.
@@ -20,7 +20,7 @@ The input tensors must have either:
 
 ## Returns
 
-A new `Tensor<i32>` with the same shape as the broadcasted inputs.
+A new `Tensor<bool>` with the same shape as the broadcasted inputs.
 
 ## Examples
 
@@ -29,7 +29,7 @@ use core::array::{ArrayTrait, SpanTrait};
 
 use orion::operators::tensor::{TensorTrait, Tensor, BoolTensor};
 
-fn and_example() -> Tensor<i32> {
+fn and_example() -> Tensor<bool> {
     let tensor_1 = TensorTrait::<bool>::new(
         shape: array![3, 3].span(), data: array![false, true, false, false, false, true, true, false, true, false, false, true].span(),
     );

docs/framework/operators/tensor/tensor.and.md

@@ -1,7 +1,7 @@
 #tensor.equal
 
 ```rust
-    fn equal(self: @Tensor<T>, other: @Tensor<T>) -> Tensor<i32>;
+    fn equal(self: @Tensor<T>, other: @Tensor<T>) -> Tensor<usize>;
 ```
 
 Check if two tensors are equal element-wise.
@@ -20,7 +20,7 @@ The input tensors must have either:
 
 ## Returns
 
-A new `Tensor<i32>` (1 if equal, 0 otherwise) with the same shape as the broadcasted inputs.
+A new `Tensor<usize>` of booleans (1 if equal, 0 otherwise) with the same shape as the broadcasted inputs.
 
 ## Examples
 
@@ -31,7 +31,7 @@ use core::array::{ArrayTrait, SpanTrait};
 
 use orion::operators::tensor::{TensorTrait, Tensor, U32Tensor};
 
-fn eq_example() -> Tensor<i32> {
+fn eq_example() -> Tensor<usize> {
     let tensor_1 = TensorTrait::<u32>::new(
         shape: array![3, 3, 3].span(), data: array![0, 1, 2, 3, 4, 5, 6, 7, 8].span(),
     );
@@ -43,7 +43,7 @@ fn eq_example() -> Tensor<i32> {
     // We can call `equal` function as follows.
     return tensor_1.equal(@tensor_2);
 }
->>> [true,true,true,true,true,false,false,false]
+>>> [1,1,1,1,1,0,0,0]
 ```
 
 Case 2: Compare tensors with different shapes
@@ -53,7 +53,7 @@ use core::array::{ArrayTrait, SpanTrait};
 
 use orion::operators::tensor::{TensorTrait, Tensor, U32Tensor};
 
-fn eq_example() -> Tensor<i32> {
+fn eq_example() -> Tensor<usize> {
     let tensor_1 = TensorTrait::<u32>::new(
         shape: array![3, 3, 3].span(), data: array![0, 1, 2, 3, 4, 5, 6, 7, 8].span(),
     );
@@ -63,5 +63,5 @@ fn eq_example() -> Tensor<i32> {
     // We can call `equal` function as follows.
     return tensor_1.equal(@tensor_2);
 }
->>> [true,true,true,false,false,false,false,false,false]
+>>> [1,1,1,0,0,0,0,0,0]
 ```

docs/framework/operators/tensor/tensor.equal.md

@@ -1,7 +1,7 @@
 #tensor.greater
 
 ```rust
-    fn greater(self: @Tensor<T>, other: @Tensor<T>) -> Tensor<i32>;
+    fn greater(self: @Tensor<T>, other: @Tensor<T>) -> Tensor<usize>;
 ```
 
 Check if each element of the first tensor is greater than the corresponding element of the second tensor.
@@ -20,7 +20,7 @@ The input tensors must have either:
 
 ## Returns
 
-A new `Tensor<i32>` of booleans (0 or 1) with the same shape as the broadcasted inputs.
+A new `Tensor<usize>` of booleans (0 or 1) with the same shape as the broadcasted inputs.
 
 ## Examples
 
@@ -31,7 +31,7 @@ use core::array::{ArrayTrait, SpanTrait};
 
 use orion::operators::tensor::{TensorTrait, Tensor, U32Tensor};
 
-fn greater_example() -> Tensor<i32> {
+fn greater_example() -> Tensor<usize> {
     let tensor_1 = TensorTrait::<u32>::new(
         shape: array![3, 3, 3].span(), data: array![0, 1, 2, 3, 4, 5, 6, 7, 8].span(),
     );
@@ -53,7 +53,7 @@ use core::array::{ArrayTrait, SpanTrait};
 
 use orion::operators::tensor::{TensorTrait, Tensor, U32Tensor};
 
-fn greater_example() -> Tensor<i32> {
+fn greater_example() -> Tensor<usize> {
     let tensor_1 = TensorTrait::<u32>::new(
         shape: array![3, 3, 3].span(), data: array![0, 1, 2, 3, 4, 5, 6, 7, 8].span(),
     );

docs/framework/operators/tensor/tensor.greater.md

@@ -1,7 +1,7 @@
 #tensor.greater_equal
 
 ```rust
-    fn greater_equal(self: @Tensor<T>, other: @Tensor<T>) -> Tensor<i32>;
+    fn greater_equal(self: @Tensor<T>, other: @Tensor<T>) -> Tensor<usize>;
 ```
 
 Check if each element of the first tensor is greater than or equal to the corresponding element of the second tensor.
@@ -31,7 +31,7 @@ use core::array::{ArrayTrait, SpanTrait};
 
 use orion::operators::tensor::{TensorTrait, Tensor, U32Tensor};
 
-fn greater_equal_example() -> Tensor<i32> {
+fn greater_equal_example() -> Tensor<usize> {
     let tensor_1 = TensorTrait::<u32>::new(
         shape: array![3, 3, 3].span(), data: array![0, 1, 2, 3, 4, 5, 6, 7, 8].span(),
     );
@@ -53,7 +53,7 @@ use core::array::{ArrayTrait, SpanTrait};
 
 use orion::operators::tensor::{TensorTrait, Tensor, U32Tensor};
 
-fn greater_equal_example() -> Tensor<i32> {
+fn greater_equal_example() -> Tensor<usize> {
     let tensor_1 = TensorTrait::<u32>::new(
         shape: array![3, 3, 3].span(), data: array![0, 1, 2, 3, 4, 5, 6, 7, 8].span(),
     );

docs/framework/operators/tensor/tensor.greater_equal.md

@@ -1,7 +1,7 @@
 ## tensor.is_inf
 
 ```rust
-   fn is_inf(self: @Tensor<T>, detect_negative: Option<u8>, detect_positive: Option<u8>) -> Tensor<usize>;
+   fn is_inf(self: @Tensor<T>, detect_negative: Option<u8>, detect_positive: Option<u8>) -> Tensor<bool>;
 ```
 
 Maps infinity to true and other values to false.
@@ -23,7 +23,7 @@ A new `Tensor<bool>` instance with entries set to true iff the input tensors cor
 use core::array::{ArrayTrait, SpanTrait};    
 use orion::operators::tensor::{BoolTensor, TensorTrait, Tensor, U32Tensor};
 
-fn is_inf_example() -> Tensor<usize> {
+fn is_inf_example() -> Tensor<bool> {
     let tensor = TensorTrait::<u32>::new(
         shape: array![6].span(), data: array![1, 0, NumberTrait::INF(), 8, NumberTrait::INF(), NumberTrait::INF()].span(),
     );

docs/framework/operators/tensor/tensor.is_inf.md

@@ -1,7 +1,7 @@
 ## tensor.is_nan
 
 ```rust
-   fn is_nan(self: @Tensor<T>) -> Tensor<usize>;
+   fn is_nan(self: @Tensor<T>) -> Tensor<bool>;
 ```
 
 Maps NaN to true and other values to false.
@@ -21,7 +21,7 @@ use core::array::{ArrayTrait, SpanTrait};
 use orion::operators::tensor::{BoolTensor, TensorTrait, Tensor, FP8x23Tensor};
 use orion::numbers::{FixedTrait, FP8x23};
 
-fn is_nan_example() -> Tensor<usize> {
+fn is_nan_example() -> Tensor<bool> {
     let mut shape = ArrayTrait::<usize>::new();
     shape.append(4);
 

docs/framework/operators/tensor/tensor.is_nan.md

@@ -1,7 +1,7 @@
 #tensor.less
 
 ```rust
-    fn less(self: @Tensor<T>, other: @Tensor<T>) -> Tensor<i32>;
+    fn less(self: @Tensor<T>, other: @Tensor<T>) -> Tensor<usize>;
 ```
 
 Check if each element of the first tensor is less than the corresponding element of the second tensor.
@@ -31,7 +31,7 @@ use core::array::{ArrayTrait, SpanTrait};
 
 use orion::operators::tensor::{TensorTrait, Tensor, U32Tensor};
 
-fn less_example() -> Tensor<i32> {
+fn less_example() -> Tensor<usize> {
     let tensor_1 = TensorTrait::<u32>::new(
         shape: array![3, 3, 3].span(), data: array![0, 1, 2, 3, 4, 5, 6, 7, 8].span(),
     );
@@ -53,7 +53,7 @@ use core::array::{ArrayTrait, SpanTrait};
 
 use orion::operators::tensor::{TensorTrait, Tensor, U32Tensor};
 
-fn less_example() -> Tensor<i32> {
+fn less_example() -> Tensor<usize> {
     let tensor_1 = TensorTrait::<u32>::new(
         shape: array![3, 3, 3].span(), data: array![0, 1, 2, 3, 4, 5, 6, 7, 8].span(),
     );
@@ -63,5 +63,5 @@ fn less_example() -> Tensor<i32> {
     // We can call `less` function as follows.
     return tensor_1.less(@tensor_2);
 }
->>> [false,false,false,false,false,false,false,true,true]
+>>> [0,0,0,0,0,0,0,1,1]
 ```

docs/framework/operators/tensor/tensor.less.md

@@ -1,7 +1,7 @@
 #tensor.less_equal
 
 ```rust
-    fn less_equal(self: @Tensor<T>, other: @Tensor<T>) -> Tensor<i32>;
+    fn less_equal(self: @Tensor<T>, other: @Tensor<T>) -> Tensor<usize>;
 ```
 
 Check if each element of the first tensor is less than or equal to the corresponding element of the second tensor.
@@ -31,7 +31,7 @@ use core::array::{ArrayTrait, SpanTrait};
 
 use orion::operators::tensor::{TensorTrait, Tensor, U32Tensor};
 
-fn less_equal_example() -> Tensor<i32> {
+fn less_equal_example() -> Tensor<usize> {
     let tensor_1 = TensorTrait::<u32>::new(
         shape: array![3, 3, 3].span(), data: array![0, 1, 2, 3, 4, 5, 6, 7, 8].span(),
     );
@@ -53,7 +53,7 @@ use core::array::{ArrayTrait, SpanTrait};
 
 use orion::operators::tensor::{TensorTrait, Tensor, U32Tensor};
 
-fn less_equal_example() -> Tensor<i32> {
+fn less_equal_example() -> Tensor<usize> {
     let tensor_1 = TensorTrait::<u32>::new(
         shape: array![3, 3, 3].span(), data: array![0, 1, 2, 3, 4, 5, 6, 7, 8].span(),
     );

docs/framework/operators/tensor/tensor.less_equal.md

@@ -1,7 +1,7 @@
 #tensor.not
 
 ```rust
-    fn not(self: @Tensor<bool>) -> Tensor<bool>;
+    fn not(self: @Tensor<bool>) -> Tensor<bool;
 ```
 
 Computes the negation of the elements in the bool type input tensor.

docs/framework/operators/tensor/tensor.not.md

@@ -1,7 +1,7 @@
 #tensor.or
 
 ```rust
-    fn or(self: @Tensor<T>, other: @Tensor<T>) -> Tensor<i32>;
+    fn or(self: @Tensor<T>, other: @Tensor<T>) -> Tensor<usize>;
 ```
 
 Computes the logical OR of two tensors element-wise.
@@ -20,7 +20,7 @@ The input tensors must have either:
 
 ## Returns
 
-A new `Tensor<i32>` (0 or 1) with the same shape as the broadcasted inputs.
+A new `Tensor<usize>` of booleans (0 or 1) with the same shape as the broadcasted inputs.
 
 ## Examples
 
@@ -31,7 +31,7 @@ use core::array::{ArrayTrait, SpanTrait};
 
 use orion::operators::tensor::{TensorTrait, Tensor, U32Tensor};
 
-fn or_example() -> Tensor<i32> {
+fn or_example() -> Tensor<usize> {
     let tensor_1 = TensorTrait::<u32>::new(
         shape: array![3, 3].span(), data: array![0, 1, 2, 3, 4, 5, 6, 7, 8].span(),
     );
@@ -52,7 +52,7 @@ use core::array::{ArrayTrait, SpanTrait};
 
 use orion::operators::tensor::{TensorTrait, Tensor, U32Tensor};
 
-fn or_example() -> Tensor<i32> {
+fn or_example() -> Tensor<usize> {
     let tensor_1 = TensorTrait::<u32>::new(
         shape: array![3, 3].span(), data: array![0, 1, 2, 3, 4, 5, 6, 7, 8].span(),
     );

docs/framework/operators/tensor/tensor.or.md

@@ -4,14 +4,13 @@
    fn reduce_sum(self: @Tensor<T>, axis: usize, keepdims: bool) -> Tensor<T>;
 ```
 
-Computes the sum of the input tensor's elements along the provided axes
+Reduces a tensor by summing its elements along a specified axis.
 
 ## Args
 
 * `self`(`@Tensor<T>`) - The input tensor.
-* `axes`(`Option<Span<usize>>`) - Optional input list of integers, along which to reduce.
-* `keepdims`(`Option<bool>`) - If true, retains reduced dimensions with length 1.
-* `noop_with_empty_axes`(`Option<bool>`) - Defines behavior if 'axes' is empty. Default behavior with 'false' is to reduce all axes. When axes is empty and this attribute is set to true, input tensor will not be reduced,and the output tensor would be equivalent to input tensor.
+* `axis`(`usize`) - The dimension to reduce.
+* `keepdims`(`bool`) - If true, retains reduced dimensions with length 1.
 
 ## Panics 
 
@@ -30,11 +29,11 @@ use orion::operators::tensor::{TensorTrait, Tensor, U32Tensor};
 
 fn reduce_sum_example() -> Tensor<u32> {
     let tensor = TensorTrait::<u32>::new(
-        shape: array![3, 2, 2].span(), data: array![1, 2, 3, 4, 5, 6, 7, 8 ,9, 10, 11, 12].span(),
+        shape: array![2, 2, 2].span(), data: array![0, 1, 2, 3, 4, 5, 6, 7].span(),
     );
 
     // We can call `reduce_sum` function as follows.
-    return tensor.reduce_sum(Option::Some(array![1].span()), Option::Some(false), Option::None);
+    return tensor.reduce_sum(axis: 0, keepdims: false);
 }
->>> [[4, 6] [12, 14] [20, 22]]
+>>> [[4,6],[8,10]]
 ```

docs/framework/operators/tensor/tensor.reduce_sum.md

@@ -1,15 +1,15 @@
 # tensor.reshape
 
 ```rust 
-   fn reshape(self: @Tensor<T>, target_shape: Span<i32>) -> Tensor<T>;
+   fn reshape(self: @Tensor<T>, target_shape: Span<usize>) -> Tensor<T>;
 ```
 
 Returns a new tensor with the specified target shape and the same data as the input tensor.
 
 ## Args
 
 * `self`(`@Tensor<T>`) - The input tensor.
-* `target_shape`(Span<i32>) - A span containing the target shape of the tensor.
+* `target_shape`(Span<usize>) - A span containing the target shape of the tensor.
 
 ## Panics