use scalar for OneHot's depth to prevent confusion (onnx#3774)

Signed-off-by: Chun-Wei Chen <[email protected]>

docs/Operators.md

@@ -12978,7 +12978,7 @@ node = onnx.helper.make_node(
 )
 indices = np.array([[1, 9],
                     [2, 4]], dtype=np.float32)
-depth = np.array([10], dtype=np.float32)
+depth = np.float32(10)
 values = np.array([off_value, on_value], dtype=output_type)
 y = one_hot(indices, depth, axis=axisValue, dtype=output_type)
 y = y * (on_value - off_value) + off_value
@@ -13004,7 +13004,7 @@ node = onnx.helper.make_node(
 )
 indices = np.array([[1, 9],
                     [2, 4]], dtype=np.float32)
-depth = np.array([10], dtype=np.float32)
+depth = np.float32(10)
 values = np.array([off_value, on_value], dtype=output_type)
 y = one_hot(indices, depth, axis=axisValue, dtype=output_type)
 y = y * (on_value - off_value) + off_value
@@ -13035,7 +13035,7 @@ indices = np.array([0, -7, -8], dtype=np.int64)
 #  [1. 1. 1. 3. 1. 1. 1. 1. 1. 1.]
 #  [1. 1. 3. 1. 1. 1. 1. 1. 1. 1.]]
 
-depth = np.array([10], dtype=np.float32)
+depth = np.float32(10)
 values = np.array([off_value, on_value], dtype=output_type)
 y = one_hot(indices, depth, axis=axisValue, dtype=output_type)
 y = y * (on_value - off_value) + off_value

docs/TestCoverage.md

@@ -8164,7 +8164,7 @@ node = onnx.helper.make_node(
 )
 indices = np.array([[1, 9],
                     [2, 4]], dtype=np.float32)
-depth = np.array([10], dtype=np.float32)
+depth = np.float32(10)
 values = np.array([off_value, on_value], dtype=output_type)
 y = one_hot(indices, depth, axis=axisValue, dtype=output_type)
 y = y * (on_value - off_value) + off_value
@@ -8188,7 +8188,7 @@ node = onnx.helper.make_node(
 )
 indices = np.array([[1, 9],
                     [2, 4]], dtype=np.float32)
-depth = np.array([10], dtype=np.float32)
+depth = np.float32(10)
 values = np.array([off_value, on_value], dtype=output_type)
 y = one_hot(indices, depth, axis=axisValue, dtype=output_type)
 y = y * (on_value - off_value) + off_value
@@ -8217,7 +8217,7 @@ indices = np.array([0, -7, -8], dtype=np.int64)
 #  [1. 1. 1. 3. 1. 1. 1. 1. 1. 1.]
 #  [1. 1. 3. 1. 1. 1. 1. 1. 1. 1.]]
 
-depth = np.array([10], dtype=np.float32)
+depth = np.float32(10)
 values = np.array([off_value, on_value], dtype=output_type)
 y = one_hot(indices, depth, axis=axisValue, dtype=output_type)
 y = y * (on_value - off_value) + off_value

onnx/backend/test/case/node/onehot.py

@@ -59,7 +59,7 @@ def export_with_axis():  # type: () -> None
         )
         indices = np.array([[1, 9],
                             [2, 4]], dtype=np.float32)
-        depth = np.array([10], dtype=np.float32)
+        depth = np.float32(10)
         values = np.array([off_value, on_value], dtype=output_type)
         y = one_hot(indices, depth, axis=axisValue, dtype=output_type)
         y = y * (on_value - off_value) + off_value
@@ -84,7 +84,7 @@ def export_with_negative_indices():  # type: () -> None
         #  [1. 1. 1. 3. 1. 1. 1. 1. 1. 1.]
         #  [1. 1. 3. 1. 1. 1. 1. 1. 1. 1.]]
 
-        depth = np.array([10], dtype=np.float32)
+        depth = np.float32(10)
         values = np.array([off_value, on_value], dtype=output_type)
         y = one_hot(indices, depth, axis=axisValue, dtype=output_type)
         y = y * (on_value - off_value) + off_value
@@ -104,7 +104,7 @@ def export_with_negative_axis():  # type: () -> None
         )
         indices = np.array([[1, 9],
                             [2, 4]], dtype=np.float32)
-        depth = np.array([10], dtype=np.float32)
+        depth = np.float32(10)
         values = np.array([off_value, on_value], dtype=output_type)
         y = one_hot(indices, depth, axis=axisValue, dtype=output_type)
         y = y * (on_value - off_value) + off_value