wehub-resource-sync
32 lines
761 B
Python
Executable File
32 lines
761 B
Python
Executable File
import unittest
|
|
import numpy as np
|
|
from prml import nn
|
|
|
|
|
|
class TestSolve(unittest.TestCase):
|
|
|
|
def test_solve(self):
|
|
A = np.array([
|
|
[2., 1.],
|
|
[1., 3.]
|
|
])
|
|
B = np.array([1., 2.])[:, None]
|
|
AinvB = np.linalg.solve(A, B)
|
|
self.assertTrue((AinvB == nn.linalg.solve(A, B).value).all())
|
|
|
|
A = nn.Parameter(A)
|
|
B = nn.Parameter(B)
|
|
for _ in range(100):
|
|
A.cleargrad()
|
|
B.cleargrad()
|
|
AinvB = nn.linalg.solve(A, B)
|
|
loss = nn.square(AinvB - 1).sum()
|
|
loss.backward()
|
|
A.value -= A.grad
|
|
B.value -= B.grad
|
|
self.assertTrue(np.allclose(AinvB.value, 1))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|