chore: import upstream snapshot with attribution
This commit is contained in:
+26
@@ -0,0 +1,26 @@
|
||||
import unittest
|
||||
import numpy as np
|
||||
from prml import nn
|
||||
|
||||
|
||||
class TestDivide(unittest.TestCase):
|
||||
|
||||
def test_divide(self):
|
||||
x = nn.Parameter(10.)
|
||||
z = x / 2
|
||||
self.assertEqual(z.value, 5)
|
||||
z.backward()
|
||||
self.assertEqual(x.grad, 0.5)
|
||||
|
||||
x = np.random.rand(5, 10, 3)
|
||||
y = np.random.rand(10, 1)
|
||||
p = nn.Parameter(y)
|
||||
z = x / p
|
||||
self.assertTrue((z.value == x / y).all())
|
||||
z.backward(np.ones((5, 10, 3)))
|
||||
d = np.sum(-x / y ** 2, axis=0).sum(axis=1, keepdims=True)
|
||||
self.assertTrue((p.grad == d).all())
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
unittest.main()
|
||||
Reference in New Issue
Block a user