-
Notifications
You must be signed in to change notification settings - Fork 0
/
step11.py
72 lines (49 loc) · 1.42 KB
/
step11.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
import numpy as np
class Variable:
def __init__(self, data):
if data is not None:
if not isinstance(data, np.ndarray):
raise TypeError(f'{(type(data))} 는 지원하지 않습니다.')
self.data = data
self.grad = None
self.creator = None
def set_creator(self, func):
self.creator = func
def backward(self):
funcs = [self.creator]
while funcs:
f = funcs.pop()
x, y = f.input, f.output
x.grad = f.backward(y.grad)
if x.creator is not None:
funcs.append(x.creator)
def as_array(x):
if np.isscalar(x):
return np.array(x)
return x
class Function:
def __call__(self, inputs):
xs = [x.data for x in inputs] # [1, 2]
ys = self.forward(xs)
outputs = [Variable(as_array(y)) for y in ys]
for output in outputs:
output.set_creator(self)
self.inputs = inputs
self.outputs = outputs
return outputs
def forward(self, xs):
raise NotImplementedError
def backward(self, gys):
raise NotImplementedError
class Add(Function):
def forward(self, xs):
x0, x1 = xs
y = x0 + x1
return (y,)
def add(xs):
return Add()(xs)
if __name__ == '__main__':
xs = [Variable(np.array(2)), Variable(np.array(3))]
ys = add(xs)
y = ys[0]
print(y.data)