Skip to content

Commit

Permalink
Batch&spatial normalization comment
Browse files Browse the repository at this point in the history
  • Loading branch information
fdwr committed Sep 12, 2024
1 parent c3a8df8 commit 2d334af
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions OperatorFormulas.html
Original file line number Diff line number Diff line change
Expand Up @@ -5705,11 +5705,14 @@ <h1 id="Operators">Operators</h1>
<td>Normalization</td>
<td>Batch&Spatial Normalization<br/>(independent channels)</td>
<td class="detailsColumn"><code>function batchAndSpatialNormalization(input, scale, bias, mean, var, epsilon, momentum)
// Generic version.
// Applies to: DML
return scale * (input - mean) / sqrt(variance + epsilon) + bias
endfunction</code>

<code>function batchAndSpatialNormalization1DBias(input, scale, bias, mean, var, epsilon, momentum)
// Assuming C == axis 1.
// Applies to: ONNX
return batchAndSpatialNormalization(input, scale, reshapeToAxes(bias, [1]), mean, var, epsilon, momentum)
endfunction</code>

Expand Down

0 comments on commit 2d334af

Please sign in to comment.