File tree Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Original file line number Diff line number Diff line change @@ -3669,9 +3669,11 @@ <h1 id="Operators">Operators</h1>
36693669< code >
36703670function broadcastDimensions(dimensionsList...)
36713671 outputRank = 0
3672- for each dimensions in dimensionsList do outputRank = max(outputRank, dimensions.size) // Determine the largest rank
3672+ for each dimensions in dimensionsList
3673+ outputRank = max(outputRank, dimensions.size) // Determine the largest rank
3674+ endfor
36733675 broadcastedDimensions = ones([outputRank]) // [1,1,...]
3674- for each dimension in dimensions // Take the max of all dimensions.
3676+ for each dimensions in dimensionsList
36753677 paddedDimensions = padLeadingValues(dimensions, outputRank, 1)
36763678 for i=0..<outputRank do
36773679 assert(paddedDimensions[i] == broadcastedDimensions[i] || paddedDimensions[i] == 1 || broadcastedDimensions[i] == 1))
You can’t perform that action at this time.
0 commit comments