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

fix matrix inverse ops st.

This commit is contained in:
linqingke 2021-01-11 10:32:32 +08:00
parent 80b46e368a
commit b9467e5340
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests/st/ops/gpu/test_matrix_inverse_op.py
View File

@ -22,6 +22,7 @@ import mindspore.nn as nn
from mindspore import Tensor
from mindspore.ops import operations as P
np.random.seed(1)
class NetMatrixInverse(nn.Cell):
def __init__(self):
@ -39,7 +40,7 @@ def test_matrix_inverse():
x0_np = np.random.uniform(-2, 2, (3, 4, 4)).astype(np.float32)
x0 = Tensor(x0_np)
expect0 = inv(x0_np)
error0 = np.ones(shape=expect0.shape) * 1.0e-5
error0 = np.ones(shape=expect0.shape) * 1.0e-3
context.set_context(mode=context.GRAPH_MODE, device_target="GPU")
matrix_inverse = NetMatrixInverse()