wehub-resource-sync
36 lines
787 B
Python
Executable File
36 lines
787 B
Python
Executable File
import unittest
|
|
import numpy as np
|
|
from prml import nn
|
|
|
|
|
|
class TestInverse(unittest.TestCase):
|
|
|
|
def test_inverse(self):
|
|
A = np.array([
|
|
[2., 1.],
|
|
[1., 3.]
|
|
])
|
|
Ainv = np.linalg.inv(A)
|
|
self.assertTrue((Ainv == nn.linalg.inv(A).value).all())
|
|
|
|
B = np.array([
|
|
[-1., 1.],
|
|
[1., 0.5]
|
|
])
|
|
A = nn.Parameter(np.array([
|
|
[-0.4, 0.7],
|
|
[0.7, 0.7]
|
|
]))
|
|
for _ in range(100):
|
|
A.cleargrad()
|
|
Ainv = nn.linalg.inv(A)
|
|
loss = nn.square(Ainv - B).sum()
|
|
loss.backward()
|
|
A.value -= 0.1 * A.grad
|
|
|
|
self.assertTrue(np.allclose(A.value, np.linalg.inv(B)))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|