Fix unidirectionally broadcast the shapes steps (#663)

huningxin · web-flow · commit 4eda710f3245 · 2024-06-05T14:56:50.000-07:00
* Fix unidirectionally broadcast the shapes steps Fix #622 * Update the call sites of "unidirectionally broadcasting the shapes" * Reverse the order of `shapeFrom` and `shapeTo`
diff --git a/index.bs b/index.bs
@@ -5903,26 +5903,26 @@ Broadcasting refers to how operations treat tensors with different shapes, and f
 
 <details open algorithm>
 <summary>
-To <dfn data-lt="unidirectionally broadcasting the shapes">unidirectionally broadcast the shapes</dfn> |shapeA| and |shapeB|, perform the following steps. |shapeA| and |shapeB| are [=/lists=] of positive integers, representing the dimensions of tensors, and the steps return a new [=/list=] of positive integers, or failure.
+To <dfn data-lt="unidirectionally broadcasting the shapes">unidirectionally broadcast the shapes</dfn> |shapeFrom| and |shapeTo|, perform the following steps. |shapeFrom| and |shapeTo| are [=/lists=] of positive integers, representing the dimensions of tensors, and the steps return a new [=/list=] of positive integers, or failure.
 </summary>
 
-1. Let |sizeA| be |shapeA|'s [=list/size=].
-1. Let |sizeB| be |shapeB|'s [=list/size=].
-1. If |sizeB| > |sizeA|, then return failure.
-1. Let |paddedB| be a [=list/clone=] of |shapeB|.
-1. While |paddedB|'s [=list/size=] is less than |sizeA|, [=list/prepend=] 1 to |paddedB|.
+1. Let |sizeFrom| be |shapeFrom|'s [=list/size=].
+1. Let |sizeTo| be |shapeTo|'s [=list/size=].
+1. If |sizeFrom| > |sizeTo|, then return failure.
+1. Let |paddedShapeFrom| be a [=list/clone=] of |shapeFrom|.
+1. While |paddedShapeFrom|'s [=list/size=] is less than |sizeTo|, [=list/prepend=] 1 to |paddedShapeFrom|.
 1. Let |outputShape| be a new [=/list=].
-1. [=list/For each=] |index| in [=the range=] 0 to |sizeA|, exclusive:
-    1. Let |dimA| be |shapeA|[|index|].
-    1. Let |dimB| be |paddedB|[|index|].
-    1. If |dimA| is not equal to |dimB| and |dimA| is not equal to 1, then return failure.
-    1. [=list/Append=] |dimA| to |outputShape|.
+1. [=list/For each=] |index| in [=the range=] 0 to |sizeTo|, exclusive:
+    1. Let |dimFrom| be |paddedShapeFrom|[|index|].
+    1. Let |dimTo| be |shapeTo|[|index|].
+    1. If |dimTo| is not equal to |dimFrom| and |dimFrom| is not equal to 1, then return failure.
+    1. [=list/Append=] |dimTo| to |outputShape|.
 1. Return |outputShape|.
 
 </details>
 
 <p algorithm>
-|shapeA| is <dfn>unidirectionally broadcastable</dfn> to |shapeB| if [=unidirectionally broadcasting the shapes=] |shapeA| and |shapeB| does not result in failure.
+|shapeFrom| is <dfn>unidirectionally broadcastable</dfn> to |shapeTo| if [=unidirectionally broadcasting the shapes=] |shapeFrom| and |shapeTo| does not result in failure.
 </p>
 
 <details open algorithm>
