AI
밑바닥부터 딥러닝3 - STEP20 - 연산자 오버로드(1)
Mul 클래스 구현 곱셈을 수행하는 클래스 Multiply class Mul(Function): def forward(self, x0, x1): y = x0 * x1 return y def backward(self, gy): x0, x1 = self.inputs[0].data, self.inputs[1].data return gy * x1, gy * x0 # 파이썬 함수로 구현 def mul(x0, x1): return Mul()(x0, x1) a = Variable(np.array(3.0)) b = Variable(np.array(2.0)) c = Variable(np.array(1.0)) y = add(mul(a, b), c) y.backward() print(y) print(a.grad) print..
2021. 8. 11. 04:44
최근댓글