Fix shape (#488)

* check in

* fix shape

* fix shape

Author: angryjinyan1

src/python/tests_extended/test_tensor_based.py

@@ -91,17 +91,17 @@ def test_datetimesplitter_complex(self):
 
         sess = set_up_onnx_model(xf, training_data)
 
-        inferencing_data = np.array([217081624, 1751241600, 217081625]).astype(np.int64).reshape(4,1)
+        inferencing_data = np.array([217081624, 1751241600, 217081625]).astype(np.int64).reshape(3,1)
         result = sess.run(None, {"tokens1": inferencing_data})
 
-        expected_years = np.array([1976, 2025, 1976]).reshape(4, 1)
-        expected_month = np.array([11, 6, 11]).reshape(4, 1)
-        expected_day = np.array([17, 30, 17]).reshape(4, 1)
-        expected_hour = np.array([12, 0, 12]).reshape(4, 1)
-        expected_minute = np.array([27, 0, 27]).reshape(4, 1)
-        expected_second = np.array([4, 0, 5]).reshape(4, 1)
-        expected_ampm = np.array([1, 0, 1]).reshape(4, 1)
-        expected_holidayname = np.array(["", "", ""]).reshape(4, 1)
+        expected_years = np.array([1976, 2025, 1976]).reshape(3, 1)
+        expected_month = np.array([11, 6, 11]).reshape(3, 1)
+        expected_day = np.array([17, 30, 17]).reshape(3, 1)
+        expected_hour = np.array([12, 0, 12]).reshape(3, 1)
+        expected_minute = np.array([27, 0, 27]).reshape(3, 1)
+        expected_second = np.array([4, 0, 5]).reshape(3, 1)
+        expected_ampm = np.array([1, 0, 1]).reshape(3, 1)
+        expected_holidayname = np.array(["", "", ""]).reshape(3, 1)
 
         np.testing.assert_array_equal(result[1],expected_years)
         np.testing.assert_array_equal(result[2],expected_month)