fix hardmax test cases make output dtype same as input (onnx#705)

aipul · Apr 4, 2018 · f0f6b3d · f0f6b3d
1 parent c970f0c
commit f0f6b3d
Show file tree

Hide file tree

Showing 10 changed files with 5 additions and 5 deletions.
diff --git a/docs/Operators.md b/docs/Operators.md
@@ -3014,7 +3014,7 @@ expect(node, inputs=[x], outputs=[y],
 
 ```python
 def hardmax_2d(x):
-    return np.eye(x.shape[1])[np.argmax(x, axis=1)]
+    return np.eye(x.shape[1], dtype=x.dtype)[np.argmax(x, axis=1)]
 
 x = np.random.randn(3, 4, 5).astype(np.float32)
 node = onnx.helper.make_node(

diff --git a/onnx/backend/test/case/node/hardmax.py b/onnx/backend/test/case/node/hardmax.py
@@ -34,7 +34,7 @@ def export():
     @staticmethod
     def export_hardmax_axis():
         def hardmax_2d(x):
-            return np.eye(x.shape[1])[np.argmax(x, axis=1)]
+            return np.eye(x.shape[1], dtype=x.dtype)[np.argmax(x, axis=1)]
 
         x = np.random.randn(3, 4, 5).astype(np.float32)
         node = onnx.helper.make_node(

diff --git a/onnx/backend/test/data/node/test_hardmax_axis_0/model.onnx b/onnx/backend/test/data/node/test_hardmax_axis_0/model.onnx
diff --git a/onnx/backend/test/data/node/test_hardmax_axis_0/test_data_set_0/output_0.pb b/onnx/backend/test/data/node/test_hardmax_axis_0/test_data_set_0/output_0.pb
diff --git a/onnx/backend/test/data/node/test_hardmax_axis_1/model.onnx b/onnx/backend/test/data/node/test_hardmax_axis_1/model.onnx
@@ -8,7 +8,7 @@
 
 b
 y
-
+
 
 
 B
diff --git a/onnx/backend/test/data/node/test_hardmax_axis_1/test_data_set_0/output_0.pb b/onnx/backend/test/data/node/test_hardmax_axis_1/test_data_set_0/output_0.pb
diff --git a/onnx/backend/test/data/node/test_hardmax_axis_2/model.onnx b/onnx/backend/test/data/node/test_hardmax_axis_2/model.onnx
@@ -8,7 +8,7 @@
 
 b
 y
-
+
 
 
 B
diff --git a/onnx/backend/test/data/node/test_hardmax_axis_2/test_data_set_0/output_0.pb b/onnx/backend/test/data/node/test_hardmax_axis_2/test_data_set_0/output_0.pb
diff --git a/onnx/backend/test/data/node/test_hardmax_default_axis/model.onnx b/onnx/backend/test/data/node/test_hardmax_default_axis/model.onnx
@@ -7,7 +7,7 @@
 
 b
 y
-
+
 
 
 B
diff --git a/onnx/backend/test/data/node/test_hardmax_default_axis/test_data_set_0/output_0.pb b/onnx/backend/test/data/node/test_hardmax_default_axis/test_data_set_0/output_0.pb