update docs and the test

titaiwangms · titaiwangms · commit 837b8066ce90 · 2026-01-27T21:04:46.000Z
diff --git a/docs/Changelog.md b/docs/Changelog.md
@@ -2177,6 +2177,9 @@ This version of the operator has been available since version 1 of the default O
 ### <a name="LpNormalization-1"></a>**LpNormalization-1**</a>
 
   Given a matrix, apply Lp-normalization along the provided axis.
+  The output is computed as: `output = input / Lp_norm(input, axis)`.
+  When the Lp norm is zero (i.e., all elements along the axis are zero),
+  the output is defined to be zero to avoid division by zero.
 
 #### Version
 
@@ -27115,6 +27118,9 @@ This version of the operator has been available since version 22 of the default
 ### <a name="LpNormalization-22"></a>**LpNormalization-22**</a>
 
   Given a matrix, apply Lp-normalization along the provided axis.
+  The output is computed as: `output = input / Lp_norm(input, axis)`.
+  When the Lp norm is zero (i.e., all elements along the axis are zero),
+  the output is defined to be zero to avoid division by zero.
 
 #### Version
 
diff --git a/docs/Operators.md b/docs/Operators.md
@@ -18187,6 +18187,9 @@ expect(
 ### <a name="LpNormalization"></a><a name="lpnormalization">**LpNormalization**</a>
 
   Given a matrix, apply Lp-normalization along the provided axis.
+  The output is computed as: `output = input / Lp_norm(input, axis)`.
+  When the Lp norm is zero (i.e., all elements along the axis are zero),
+  the output is defined to be zero to avoid division by zero.
 
 #### Version
 
@@ -18307,7 +18310,8 @@ x = np.array(
     dtype=np.float32,
 )
 l2_norm_axis_0 = np.sqrt(np.sum(x**2, axis=0, keepdims=True))
-y = x / l2_norm_axis_0
+# When norm is 0, output is 0 (0/0 = 0)
+y = np.where(l2_norm_axis_0 == 0, 0, x / l2_norm_axis_0)
 expect(node, inputs=[x], outputs=[y], name="test_l2normalization_axis_0")
 ```
 
diff --git a/docs/TestCoverage.md b/docs/TestCoverage.md
@@ -13004,7 +13004,8 @@ x = np.array(
     dtype=np.float32,
 )
 l2_norm_axis_0 = np.sqrt(np.sum(x**2, axis=0, keepdims=True))
-y = x / l2_norm_axis_0
+# When norm is 0, output is 0 (0/0 = 0)
+y = np.where(l2_norm_axis_0 == 0, 0, x / l2_norm_axis_0)
 expect(node, inputs=[x], outputs=[y], name="test_l2normalization_axis_0")
 ```
 
diff --git a/onnx/backend/test/data/node/test_l2normalization_axis_0/test_data_set_0/output_0.pb b/onnx/backend/test/data/node/test_l2normalization_axis_0/test_data_set_0/output_0.pb

Original file line number	Diff line number	Diff line change
`@@ -13004,7 +13004,8 @@ x = np.array(`
`13004`	`13004`	`dtype=np.float32,`
`13005`	`13005`	`)`
`13006`	`13006`	`l2_norm_axis_0 = np.sqrt(np.sum(x**2, axis=0, keepdims=True))`
`13007`		`-y = x / l2_norm_axis_0`
	`13007`	`+# When norm is 0, output is 0 (0/0 = 0)`
	`13008`	`+y = np.where(l2_norm_axis_0 == 0, 0, x / l2_norm_axis_0)`
`13008`	`13009`	`expect(node, inputs=[x], outputs=[y], name="test_l2normalization_axis_0")`
`13009`	`13010`	```
`13010`	`13011`