From bf978fffe5ee8b68882336aaf8f6c4be9951de06 Mon Sep 17 00:00:00 2001 From: zhupengyang <1165938320@qq.com> Date: Sat, 4 Jan 2020 23:10:01 +0800 Subject: [PATCH] all cases use large shape (#22084) flatten, flatten2, gather, gather_nd, group_norm, gru, gru_unit, reduce_max, reduce_min, reduce_mean, reduce_sum, reshape, scatter_nd_add, sequence_expand, sequence_expand_as, sequence_pad --- .../sequence/test_sequence_expand.py | 14 ++--- .../sequence/test_sequence_expand_as.py | 6 +- .../sequence/test_sequence_pad_op.py | 20 +++---- .../fluid/tests/unittests/test_flatten2_op.py | 4 +- .../fluid/tests/unittests/test_flatten_op.py | 4 +- .../tests/unittests/test_gather_nd_op.py | 6 +- .../fluid/tests/unittests/test_gather_op.py | 2 +- .../tests/unittests/test_group_norm_op.py | 2 +- .../fluid/tests/unittests/test_gru_op.py | 7 +-- .../fluid/tests/unittests/test_gru_unit_op.py | 2 +- .../fluid/tests/unittests/test_reduce_op.py | 12 ++-- .../fluid/tests/unittests/test_reshape_op.py | 58 +++++++++---------- .../tests/unittests/test_scatter_nd_op.py | 8 +-- 13 files changed, 67 insertions(+), 78 deletions(-) diff --git a/python/paddle/fluid/tests/unittests/sequence/test_sequence_expand.py b/python/paddle/fluid/tests/unittests/sequence/test_sequence_expand.py index 51e6f31d9c..b3d877a0cd 100644 --- a/python/paddle/fluid/tests/unittests/sequence/test_sequence_expand.py +++ b/python/paddle/fluid/tests/unittests/sequence/test_sequence_expand.py @@ -82,7 +82,7 @@ class TestSequenceExpand(OpTest): class TestSequenceExpandCase1(TestSequenceExpand): def set_data(self): - x_data = np.random.uniform(0.1, 1, [5, 1]).astype('float64') + x_data = np.random.uniform(0.1, 1, [5, 20]).astype('float64') y_data = np.random.uniform(0.1, 1, [13, 1]).astype('float64') y_lod = [[2, 3], [2, 2, 3, 3, 3]] self.inputs = {'X': x_data, 'Y': (y_data, y_lod)} @@ -91,7 +91,7 @@ class TestSequenceExpandCase1(TestSequenceExpand): class TestSequenceExpandCase2(TestSequenceExpand): def set_data(self): - x_data = np.random.uniform(0.1, 1, [1, 2, 2]).astype('float64') + x_data = np.random.uniform(0.1, 1, [1, 2, 50]).astype('float64') x_lod = [[1]] y_data = np.random.uniform(0.1, 1, [2, 2, 2]).astype('float64') y_lod = [[2], [1, 1]] @@ -101,7 +101,7 @@ class TestSequenceExpandCase2(TestSequenceExpand): class TestSequenceExpandCase3(TestSequenceExpand): def set_data(self): - x_data = np.random.uniform(0.1, 1, [4, 1]).astype('float64') + x_data = np.random.uniform(0.1, 1, [4, 25]).astype('float64') x_lod = [[1, 1, 1, 1]] y_data = np.random.uniform(0.1, 1, [8, 1]).astype('float64') y_lod = [[2, 2, 2, 2]] @@ -110,8 +110,8 @@ class TestSequenceExpandCase3(TestSequenceExpand): class TestSequenceExpandCase4(TestSequenceExpand): def set_data(self): - data = np.random.uniform(0.1, 1, [5 * 2, 1]) - x_data = np.array(data).reshape([5, 2]).astype('float64') + data = np.random.uniform(0.1, 1, [5 * 20, 1]) + x_data = np.array(data).reshape([5, 20]).astype('float64') x_lod = [[2, 3]] y_data = np.random.uniform(0.1, 1, [5, 1]).astype('float64') y_lod = [[2], [2, 3]] @@ -120,7 +120,7 @@ class TestSequenceExpandCase4(TestSequenceExpand): class TestSequenceExpandCase5(TestSequenceExpand): def set_data(self): - x_data = np.random.uniform(0.1, 1, [6, 1]).astype('float64') + x_data = np.random.uniform(0.1, 1, [6, 20]).astype('float64') y_data = np.random.uniform(0.1, 1, [13, 1]).astype('float64') y_lod = [[2, 4], [2, 2, 3, 0, 3, 3]] self.inputs = {'X': x_data, 'Y': (y_data, y_lod)} @@ -129,7 +129,7 @@ class TestSequenceExpandCase5(TestSequenceExpand): class TestSequenceExpandCase6(TestSequenceExpand): def set_data(self): - x_data = np.random.uniform(0.1, 1, [4, 1]).astype('float64') + x_data = np.random.uniform(0.1, 1, [4, 25]).astype('float64') x_lod = [[1, 1, 0, 1, 1]] y_data = np.random.uniform(0.1, 1, [8, 1]).astype('float64') y_lod = [[0, 2, 4, 2, 0]] diff --git a/python/paddle/fluid/tests/unittests/sequence/test_sequence_expand_as.py b/python/paddle/fluid/tests/unittests/sequence/test_sequence_expand_as.py index f038b00a3b..3b3a1e9d5c 100644 --- a/python/paddle/fluid/tests/unittests/sequence/test_sequence_expand_as.py +++ b/python/paddle/fluid/tests/unittests/sequence/test_sequence_expand_as.py @@ -59,7 +59,7 @@ class TestSequenceExpandAs(OpTest): class TestSequenceExpandAsCase1(TestSequenceExpandAs): def set_data(self): - x_data = np.random.uniform(0.1, 1, [5, 1]).astype('float64') + x_data = np.random.uniform(0.1, 1, [5, 20]).astype('float64') x_lod = [[2, 3]] y_data = np.random.uniform(0.1, 1, [10, 1]).astype('float64') y_lod = [[2, 2, 0, 3, 3]] @@ -68,7 +68,7 @@ class TestSequenceExpandAsCase1(TestSequenceExpandAs): class TestSequenceExpandAsCase2(TestSequenceExpandAs): def set_data(self): - x_data = np.random.uniform(0.1, 1, [5, 1]).astype('float64') + x_data = np.random.uniform(0.1, 1, [5, 20]).astype('float64') x_lod = [[2, 3]] y_data = np.random.uniform(0.1, 1, [10, 1]).astype('float64') y_lod = [[0, 4, 0, 6, 0]] @@ -77,7 +77,7 @@ class TestSequenceExpandAsCase2(TestSequenceExpandAs): class TestSequenceExpandAsCase3(TestSequenceExpandAs): def set_data(self): - x_data = np.random.uniform(0.1, 1, [1, 2, 2]).astype('float64') + x_data = np.random.uniform(0.1, 1, [1, 2, 50]).astype('float64') x_lod = [[1]] y_data = np.random.uniform(0.1, 1, [2, 2, 2]).astype('float64') y_lod = [[2]] diff --git a/python/paddle/fluid/tests/unittests/sequence/test_sequence_pad_op.py b/python/paddle/fluid/tests/unittests/sequence/test_sequence_pad_op.py index 83c28edd23..d04f10139b 100644 --- a/python/paddle/fluid/tests/unittests/sequence/test_sequence_pad_op.py +++ b/python/paddle/fluid/tests/unittests/sequence/test_sequence_pad_op.py @@ -82,16 +82,16 @@ class TestSequencePadOp(OpTest): class TestSequencePadOp2(TestSequencePadOp): def set_attr(self): - self.x_shape = [12, 4] + self.x_shape = [12, 10] self.x_len_lod = [[2, 3, 4, 3]] - self.pad_value = [1.0, 2.0, 3.0, 4.0] + self.pad_value = np.random.random((10)) self.padded_length = -1 self.dtype = 'float64' class TestSequencePadOp3(TestSequencePadOp): def set_attr(self): - self.x_shape = [12, 4] + self.x_shape = [12, 10] self.x_len_lod = [[2, 3, 4, 3]] self.pad_value = [1.0] self.padded_length = 7 @@ -100,16 +100,16 @@ class TestSequencePadOp3(TestSequencePadOp): class TestSequencePadOp4(TestSequencePadOp): def set_attr(self): - self.x_shape = [12, 4] + self.x_shape = [12, 10] self.x_len_lod = [[2, 3, 4, 3]] - self.pad_value = [1.0, 2.0, 3.0, 4.0] + self.pad_value = np.random.random((10)) self.padded_length = 7 self.dtype = 'float64' class TestSequencePadOp5(TestSequencePadOp): def set_attr(self): - self.x_shape = [12, 2, 2] + self.x_shape = [12, 2, 5] self.x_len_lod = [[2, 3, 4, 3]] self.pad_value = [1.0] self.padded_length = -1 @@ -118,16 +118,16 @@ class TestSequencePadOp5(TestSequencePadOp): class TestSequencePadOp6(TestSequencePadOp): def set_attr(self): - self.x_shape = [12, 2, 2] + self.x_shape = [12, 2, 5] self.x_len_lod = [[2, 3, 4, 3]] - self.pad_value = [[1.0, 2.0], [3.0, 4.0]] + self.pad_value = np.random.random((2, 5)) self.padded_length = -1 self.dtype = 'float64' class TestSequencePadOp7(TestSequencePadOp): def set_attr(self): - self.x_shape = [12, 2, 2] + self.x_shape = [12, 2, 5] self.x_len_lod = [[2, 3, 4, 3]] self.pad_value = [1.0] self.padded_length = 7 @@ -136,7 +136,7 @@ class TestSequencePadOp7(TestSequencePadOp): class TestSequencePadOp8(TestSequencePadOp): def set_attr(self): - self.x_shape = [12, 2, 2] + self.x_shape = [12, 2, 5] self.x_len_lod = [[0, 8, 0, 4, 0]] self.pad_value = [1.0] self.padded_length = 10 diff --git a/python/paddle/fluid/tests/unittests/test_flatten2_op.py b/python/paddle/fluid/tests/unittests/test_flatten2_op.py index 9fe2d7b20c..c37e115fdc 100644 --- a/python/paddle/fluid/tests/unittests/test_flatten2_op.py +++ b/python/paddle/fluid/tests/unittests/test_flatten2_op.py @@ -55,8 +55,8 @@ class TestFlattenOp(TestFlattenOp): class TestFlattenOpWithDefaultAxis(TestFlattenOp): def init_test_case(self): - self.in_shape = (3, 2, 2, 3) - self.new_shape = (3, 12) + self.in_shape = (10, 2, 2, 3) + self.new_shape = (10, 12) def init_attrs(self): self.attrs = {} diff --git a/python/paddle/fluid/tests/unittests/test_flatten_op.py b/python/paddle/fluid/tests/unittests/test_flatten_op.py index 902ce1613a..cf568970b7 100644 --- a/python/paddle/fluid/tests/unittests/test_flatten_op.py +++ b/python/paddle/fluid/tests/unittests/test_flatten_op.py @@ -52,8 +52,8 @@ class TestFlattenOp(TestFlattenOp): class TestFlattenOpWithDefaultAxis(TestFlattenOp): def init_test_case(self): - self.in_shape = (3, 2, 2, 3) - self.new_shape = (3, 12) + self.in_shape = (10, 2, 2, 3) + self.new_shape = (10, 12) def init_attrs(self): self.attrs = {} diff --git a/python/paddle/fluid/tests/unittests/test_gather_nd_op.py b/python/paddle/fluid/tests/unittests/test_gather_nd_op.py index 95cfdaa5e8..892f63bf15 100644 --- a/python/paddle/fluid/tests/unittests/test_gather_nd_op.py +++ b/python/paddle/fluid/tests/unittests/test_gather_nd_op.py @@ -47,8 +47,7 @@ class TestGatherNdOpWithLowIndex(OpTest): def setUp(self): self.op_type = "gather_nd" - xnp = np.array( - [[65, 17, 2], [14, 25, 1], [76, 22, 3]]).astype("float64") + xnp = np.random.uniform(0, 100, (10, 10)).astype("float64") index = np.array([[1], [2]]).astype("int64") self.inputs = {'X': xnp, 'Index': index} @@ -69,8 +68,7 @@ class TestGatherNdOpWithSameIndexAsX(OpTest): def setUp(self): self.op_type = "gather_nd" - xnp = np.array( - [[65, 17, 2], [14, 25, 1], [76, 22, 3]]).astype("float64") + xnp = np.random.uniform(0, 100, (10, 10)).astype("float64") index = np.array([[1, 1], [2, 1]]).astype("int64") self.inputs = {'X': xnp, 'Index': index} diff --git a/python/paddle/fluid/tests/unittests/test_gather_op.py b/python/paddle/fluid/tests/unittests/test_gather_op.py index 9f89bd5b5c..340be3238d 100644 --- a/python/paddle/fluid/tests/unittests/test_gather_op.py +++ b/python/paddle/fluid/tests/unittests/test_gather_op.py @@ -62,7 +62,7 @@ class TestCase2(TestGatherOp): """ For int64_t index type """ - self.x_shape = (10) + self.x_shape = (100) self.x_type = "float64" self.index = [1, 3, 5] self.index_type = "int64" diff --git a/python/paddle/fluid/tests/unittests/test_group_norm_op.py b/python/paddle/fluid/tests/unittests/test_group_norm_op.py index eb69e283a3..716019f3d1 100644 --- a/python/paddle/fluid/tests/unittests/test_group_norm_op.py +++ b/python/paddle/fluid/tests/unittests/test_group_norm_op.py @@ -45,7 +45,7 @@ class TestGroupNormOp(OpTest): self.op_type = "group_norm" self.data_format = "NCHW" self.dtype = np.float64 - self.shape = (2, 4, 3, 5) + self.shape = (2, 100, 3, 5) self.attrs = {'epsilon': 1e-5, 'groups': 2, 'data_layout': "NCHW"} self.compare_between_place = False self.init_test_case() diff --git a/python/paddle/fluid/tests/unittests/test_gru_op.py b/python/paddle/fluid/tests/unittests/test_gru_op.py index cc041991e7..b3a6d5cc7c 100644 --- a/python/paddle/fluid/tests/unittests/test_gru_op.py +++ b/python/paddle/fluid/tests/unittests/test_gru_op.py @@ -107,7 +107,7 @@ class TestGRUOp(OpTest): def setUp(self): self.op_type = "gru" self.lod = [[2, 4, 3]] - self.D = 5 + self.D = 40 self.is_reverse = False self.with_h0 = True self.with_bias = True @@ -169,27 +169,23 @@ class TestGRUOriginMode(TestGRUOp): class TestGRUOp2(TestGRUOp): def set_confs(self): - self.D = 19 self.dtype = 'float64' class TestGRUOp2Len0(TestGRUOp): def set_confs(self): - self.D = 19 self.lod = [[2, 0, 4]] self.dtype = 'float64' class TestGRUOp2OriginMode(TestGRUOp): def set_confs(self): - self.D = 19 self.dtype = 'float64' self.origin_mode = True class TestGRUOp2OriginModeLen0(TestGRUOp): def set_confs(self): - self.D = 19 self.lod = [[0, 3, 4]] self.dtype = 'float64' self.origin_mode = True @@ -197,7 +193,6 @@ class TestGRUOp2OriginModeLen0(TestGRUOp): class TestGRUOp2OriginModeLastLen0(TestGRUOp): def set_confs(self): - self.D = 19 self.lod = [[0, 3, 0]] self.dtype = 'float64' self.origin_mode = True diff --git a/python/paddle/fluid/tests/unittests/test_gru_unit_op.py b/python/paddle/fluid/tests/unittests/test_gru_unit_op.py index 78f2f030f5..6164591992 100644 --- a/python/paddle/fluid/tests/unittests/test_gru_unit_op.py +++ b/python/paddle/fluid/tests/unittests/test_gru_unit_op.py @@ -45,7 +45,7 @@ def relu(x): class TestGRUUnitOp(OpTest): batch_size = 5 - frame_size = 10 + frame_size = 40 activate = { GRUActivationType.identity: identity, GRUActivationType.sigmoid: sigmoid, diff --git a/python/paddle/fluid/tests/unittests/test_reduce_op.py b/python/paddle/fluid/tests/unittests/test_reduce_op.py index 2a1f8d3d55..efd1b15e92 100644 --- a/python/paddle/fluid/tests/unittests/test_reduce_op.py +++ b/python/paddle/fluid/tests/unittests/test_reduce_op.py @@ -333,7 +333,7 @@ class TestKeepDimReduceSumMultiAxises(OpTest): class TestReduceSumWithDimOne(OpTest): def setUp(self): self.op_type = "reduce_sum" - self.inputs = {'X': np.random.random((10, 1, 1)).astype("float64")} + self.inputs = {'X': np.random.random((100, 1, 1)).astype("float64")} self.attrs = {'dim': [1, 2], 'keep_dim': True} self.outputs = { 'Out': self.inputs['X'].sum(axis=tuple(self.attrs['dim']), @@ -350,7 +350,7 @@ class TestReduceSumWithDimOne(OpTest): class TestReduceSumWithNumelOne(OpTest): def setUp(self): self.op_type = "reduce_sum" - self.inputs = {'X': np.random.random((1, 1)).astype("float64")} + self.inputs = {'X': np.random.random((100, 1)).astype("float64")} self.attrs = {'dim': [1], 'keep_dim': False} self.outputs = { 'Out': self.inputs['X'].sum(axis=tuple(self.attrs['dim']), @@ -367,7 +367,7 @@ class TestReduceSumWithNumelOne(OpTest): class TestReduceMeanWithDimOne(OpTest): def setUp(self): self.op_type = "reduce_mean" - self.inputs = {'X': np.random.random((10, 1, 1)).astype("float64")} + self.inputs = {'X': np.random.random((100, 1, 1)).astype("float64")} self.attrs = {'dim': [1], 'keep_dim': False} self.outputs = { 'Out': self.inputs['X'].mean( @@ -384,7 +384,7 @@ class TestReduceMeanWithDimOne(OpTest): class TestReduceMeanWithNumelOne(OpTest): def setUp(self): self.op_type = "reduce_mean" - self.inputs = {'X': np.random.random((1, 1)).astype("float64")} + self.inputs = {'X': np.random.random((100, 1)).astype("float64")} self.attrs = {'dim': [1], 'keep_dim': True} self.outputs = { 'Out': self.inputs['X'].mean( @@ -401,7 +401,7 @@ class TestReduceMeanWithNumelOne(OpTest): class TestReduceAll(OpTest): def setUp(self): self.op_type = "reduce_sum" - self.inputs = {'X': np.random.random((1, 1, 1)).astype("float64")} + self.inputs = {'X': np.random.random((100, 1, 1)).astype("float64")} self.attrs = {'reduce_all': True, 'keep_dim': False} self.outputs = {'Out': self.inputs['X'].sum()} @@ -415,7 +415,7 @@ class TestReduceAll(OpTest): class Test1DReduceWithAxes1(OpTest): def setUp(self): self.op_type = "reduce_sum" - self.inputs = {'X': np.random.random(1).astype("float64")} + self.inputs = {'X': np.random.random(100).astype("float64")} self.attrs = {'dim': [0], 'keep_dim': False} self.outputs = {'Out': self.inputs['X'].sum(axis=0)} diff --git a/python/paddle/fluid/tests/unittests/test_reshape_op.py b/python/paddle/fluid/tests/unittests/test_reshape_op.py index b7c3b52c0e..826737aeef 100644 --- a/python/paddle/fluid/tests/unittests/test_reshape_op.py +++ b/python/paddle/fluid/tests/unittests/test_reshape_op.py @@ -55,9 +55,9 @@ class TestReshapeOpDimInfer1(TestReshapeOp): class TestReshapeOpDimInfer2(TestReshapeOp): def init_data(self): - self.ori_shape = (2, 2, 6) - self.new_shape = (2, 0, 3, -1) - self.infered_shape = (2, 2, 3, -1) + self.ori_shape = (10, 2, 6) + self.new_shape = (10, 0, 3, -1) + self.infered_shape = (10, 2, 3, -1) # situation 2: have shape(list, no tensor), have actual shape(Tensor) @@ -78,9 +78,9 @@ class TestReshapeOpWithInputShape(OpTest): } def init_data(self): - self.ori_shape = (6, 5) - self.new_shape = (0, -1, 5) - self.actual_shape = (2, 3, 5) + self.ori_shape = (6, 20) + self.new_shape = (0, -1, 20) + self.actual_shape = (2, 3, 20) def test_check_output(self): self.check_output(no_check_set=['XShape']) @@ -111,9 +111,9 @@ class TestReshapeOp_attr_ShapeTensor(OpTest): } def init_data(self): - self.ori_shape = (2, 25) - self.new_shape = (5, 10) - self.infered_shape = (5, 10) + self.ori_shape = (4, 25) + self.new_shape = (10, 10) + self.infered_shape = (10, 10) self.shape = (-1, -1) def test_check_output(self): @@ -125,18 +125,18 @@ class TestReshapeOp_attr_ShapeTensor(OpTest): class TestReshapeOpDimInfer1_attr_ShapeTensor(TestReshapeOp_attr_ShapeTensor): def init_data(self): - self.ori_shape = (5, 10) - self.new_shape = (5, -1, 5) - self.infered_shape = (5, -1, 5) + self.ori_shape = (5, 20) + self.new_shape = (5, -1, 20) + self.infered_shape = (5, -1, 20) self.shape = (5, -1, -1) class TestReshapeOpDimInfer2_attr_ShapeTensor(TestReshapeOp_attr_ShapeTensor): def init_data(self): - self.ori_shape = (2, 2, 6) - self.new_shape = (2, 0, 3, -1) - self.infered_shape = (2, 2, 3, -1) - self.shape = (2, 0, 3, -1) + self.ori_shape = (10, 2, 6) + self.new_shape = (10, 0, 3, -1) + self.infered_shape = (10, 2, 3, -1) + self.shape = (10, 0, 3, -1) # Situation 4: have shape(Tensor), no actual shape(Tensor) @@ -157,9 +157,9 @@ class TestReshapeOp_attr_OnlyShape(OpTest): } def init_data(self): - self.ori_shape = (2, 25) - self.new_shape = (5, 10) - self.infered_shape = (5, 10) + self.ori_shape = (4, 25) + self.new_shape = (10, 10) + self.infered_shape = (10, 10) def test_check_output(self): self.check_output(no_check_set=['XShape']) @@ -170,18 +170,18 @@ class TestReshapeOp_attr_OnlyShape(OpTest): class TestReshapeOpDimInfer1_attr_OnlyShape(TestReshapeOp_attr_OnlyShape): def init_data(self): - self.ori_shape = (5, 10) - self.new_shape = (5, -1, 5) - self.infered_shape = (5, -1, 5) + self.ori_shape = (5, 20) + self.new_shape = (5, -1, 10) + self.infered_shape = (5, -1, 10) self.shape = (5, -1, -1) class TestReshapeOpDimInfer2_attr_OnlyShape(TestReshapeOp_attr_OnlyShape): def init_data(self): - self.ori_shape = (2, 2, 6) - self.new_shape = (2, 0, 3, -1) - self.infered_shape = (2, 2, 3, -1) - self.shape = (2, 0, 3, -1) + self.ori_shape = (10, 2, 6) + self.new_shape = (10, 0, 3, -1) + self.infered_shape = (10, 2, 3, -1) + self.shape = (10, 0, 3, -1) # test int8 data type on CPU @@ -207,9 +207,9 @@ class TestReshapeInt8Op(OpTest): self.dtype = np.int8 def init_data(self): - self.ori_shape = (2, 2, 6) - self.new_shape = (2, 0, 3, -1) - self.infered_shape = (2, 2, 3, -1) + self.ori_shape = (10, 2, 6) + self.new_shape = (10, 0, 3, -1) + self.infered_shape = (10, 2, 3, -1) def test_check_output(self): self.check_output_with_place( diff --git a/python/paddle/fluid/tests/unittests/test_scatter_nd_op.py b/python/paddle/fluid/tests/unittests/test_scatter_nd_op.py index c8f4ed24e5..7efa3e980c 100644 --- a/python/paddle/fluid/tests/unittests/test_scatter_nd_op.py +++ b/python/paddle/fluid/tests/unittests/test_scatter_nd_op.py @@ -65,12 +65,10 @@ class TestScatterNdAddSimpleOp(OpTest): def setUp(self): self.op_type = "scatter_nd_add" - #ref_np = np.array([0, 1, 2, 3, 4, 5, 6, 7, 8]).astype("float32") ref_np = np.random.random([100]).astype("float64") index_np = np.random.randint(0, 100, [100, 1]).astype("int32") updates_np = np.random.random([100]).astype("float64") expect_np = numpy_scatter_nd_add(ref_np.copy(), index_np, updates_np) - #expect_np = [ 0. 23. 12. 14. 4. 17. 6. 7. 8.] self.inputs = {'X': ref_np, 'Index': index_np, 'Updates': updates_np} self.outputs = {'Out': expect_np} @@ -89,13 +87,11 @@ class TestScatterNdAddWithEmptyIndex(OpTest): def setUp(self): self.op_type = "scatter_nd_add" - ref_np = np.array([[65, 17], [-14, -25]]).astype("float64") + ref_np = np.random.random((10, 10)).astype("float64") index_np = np.array([[], []]).astype("int32") - updates_np = np.array([[[-1, -2], [1, 2]], - [[3, 4], [-3, -4]]]).astype("float64") + updates_np = np.random.random((2, 10, 10)).astype("float64") expect_np = numpy_scatter_nd_add(ref_np.copy(), index_np, updates_np) - #expect_np = [[67, 19], [-16, -27]] self.inputs = {'X': ref_np, 'Index': index_np, 'Updates': updates_np} self.outputs = {'Out': expect_np} -- GitLab