先学计算图,从计算图中可以看到正向传播和反向传播
每个苹果100元,每个橘子150,最后有个0.1的消费税 price*(1+0.1)
细线是正向,粗线是反向
写在细线上面的数是正向传播,写在粗线下面的是反向传播(也就是求导)
z = x*y
z 对x 求导等于y
z对y 求导等于 x
反向传播给的1的含义是,总价格上涨1元,那么前面输入 应该做出“多大努力”
buy_apple_orange
from MulLayer import *
from AddLayer import *
apple = 100
apple_num = 2
orange = 150
orange_num = 3
tax = 1.1
mul_apple_Layer = MulLayer()
mul_orange_Layer = MulLayer()
add_apple_orange_layer = AddLayer()
mul_tax_Layer = MulLayer()
#forward
apple_price = mul_apple_Layer.forward(apple,apple_num)
orange_price = mul_orange_Layer.forward(orange,orange_num)
all_price = add_apple_orange_layer.forward(apple_price,orange_price)
price = mul_tax_Layer.forward(all_price,tax)
print(price)
#backward
dprice = 1
dall_price,dtax = mul_tax_Layer.backward(dprice)
dapple_price,dorange_price = add_apple_orange_layer.backward(dall_price)
dapple ,dapple_num = mul_apple_Layer.backward(dapple_price)
dorange,dorange_num = mul_orange_Layer.backward(dorange_price)
print(dapple_num,dapple,dorange,dorange_num,dtax)
addlayer
class AddLayer:
def __init__(self):
pass
def forward(self,x,y):
out = x+y
return out
def backward(self,dout):
dx = dout * 1
dy = dout * 1
return dx ,dy
mul_layer
class MulLayer:
def __init__(self):
self.x = None
self.y = None
def forward(self,x,y):
self.x = x
self.y = y
out = x * y
return out
def backward(self,dout):
dx = dout * self.y
dy = dout * self.x
return dx,dy