Add random seed for SparseSegmentMean test case.

2022-06-09 17:43:37 +08:00 · 2022-06-09 17:43:37 +08:00 · 17653575c5
parent 4875e3dfc8
commit 17653575c5
1 changed files with 1 additions and 0 deletions
--- a/tests/st/ops/gpu/test_sparse_segment_mean_op.py
+++ b/tests/st/ops/gpu/test_sparse_segment_mean_op.py
@ -52,6 +52,7 @@ def test_net(data_type, index_type, error):
    Description:  test cases for SparseSegmentMean operator.
    Expectation: the result match expectation.
    """
+    np.random.seed(0)
    data_shape = (4, 5, 6)
    index_shape = (5,)
    x = np.random.random(size=data_shape).astype(data_type)