wehub-resource-sync
28 lines
641 B
Python
Executable File
28 lines
641 B
Python
Executable File
import unittest
|
|
import numpy as np
|
|
from prml import nn
|
|
|
|
|
|
class TestLogdet(unittest.TestCase):
|
|
|
|
def test_logdet(self):
|
|
A = np.array([
|
|
[2., 1.],
|
|
[1., 3.]
|
|
])
|
|
logdetA = np.linalg.slogdet(A)[1]
|
|
self.assertTrue((logdetA == nn.linalg.logdet(A).value).all())
|
|
|
|
A = nn.Parameter(A)
|
|
for _ in range(100):
|
|
A.cleargrad()
|
|
logdetA = nn.linalg.logdet(A)
|
|
loss = nn.square(logdetA - 1)
|
|
loss.backward()
|
|
A.value -= 0.1 * A.grad
|
|
self.assertAlmostEqual(logdetA.value, 1)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|