From e89fbd610591cb8b9074ddcd0e982a96179f84a2 Mon Sep 17 00:00:00 2001
From: chachaleo <charlotte.leo@hotmail.com>
Date: Tue, 24 Oct 2023 15:48:08 +0700
Subject: [PATCH] modification function broadcast_shape

---
 src/operators/tensor/helpers.cairo | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/operators/tensor/helpers.cairo b/src/operators/tensor/helpers.cairo
index fe08cdbb5..e72b08a2e 100644
--- a/src/operators/tensor/helpers.cairo
+++ b/src/operators/tensor/helpers.cairo
@@ -276,7 +276,7 @@ fn broadcast_shape(mut shape1: Span<usize>, mut shape2: Span<usize>) -> Span<usi
         let mut dim1 = 1;
         let mut dim2 = 1;
 
-        match shape1.pop_back() {
+        match shape1.pop_front() {
             Option::Some(item) => {
                 dim1 = *item;
             },
@@ -287,7 +287,7 @@ fn broadcast_shape(mut shape1: Span<usize>, mut shape2: Span<usize>) -> Span<usi
             }
         };
 
-        match shape2.pop_back() {
+        match shape2.pop_front() {
             Option::Some(item) => {
                 dim2 = *item;
             },
@@ -302,7 +302,7 @@ fn broadcast_shape(mut shape1: Span<usize>, mut shape2: Span<usize>) -> Span<usi
         result.append(broadcasted_dim);
     };
 
-    return result.reverse().span();
+    return result.span();
 }