Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

matrix_power:remove the test case for n < 0

This commit is contained in:
guozhibin 2023-02-16 15:01:44 +08:00
parent 4a8d98014d
commit 1dcab90aa5
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/st/ops/test_func_matrix_power.py
View File

@ -44,7 +44,7 @@ def test_matrix_power(mode):
net_matrix_power = NetMatrixPower() net_matrix_power = NetMatrixPower()
for arr in arrs: for arr in arrs:
for n in range(-2, 4): for n in range(0, 4):
expect_out = np.linalg.matrix_power(arr, n) expect_out = np.linalg.matrix_power(arr, n)
out = net_matrix_power(ms.Tensor(arr), n) out = net_matrix_power(ms.Tensor(arr), n)
assert np.allclose(out.asnumpy(), expect_out, rtol=1e-4, atol=1e-4) assert np.allclose(out.asnumpy(), expect_out, rtol=1e-4, atol=1e-4)

2
tests/st/tensor/test_matrix_power.py
View File

@ -43,7 +43,7 @@ def test_matrix_power(mode):
net_matrix_power = NetMatrixPower() net_matrix_power = NetMatrixPower()
for arr in arrs: for arr in arrs:
for n in range(-2, 4): for n in range(0, 4):
expect_out = np.linalg.matrix_power(arr, n) expect_out = np.linalg.matrix_power(arr, n)
out = net_matrix_power(ms.Tensor(arr), n) out = net_matrix_power(ms.Tensor(arr), n)
assert np.allclose(out.asnumpy(), expect_out, rtol=1e-4, atol=1e-4) assert np.allclose(out.asnumpy(), expect_out, rtol=1e-4, atol=1e-4)