深度学习入门——加法层（Addlayer）

• 上游的值直接传给下游
  

class ADDlayer:
    def __init__(self):
        pass # 什么也不运行

    def forward(self, x, y):
        out = x + y
        return out

    def backward(self, dout):
        dx = dout
        dy = dout

        return dx, dy


# -----------------------------买苹果的例子，正向传播-----------------------------------------------------
apple = 100
apple_num = 2
tax = 1.1

# layer
mul_apple_layer = ADDlayer()
mul_tax_layer = ADDlayer()

# forward
apple_price = mul_apple_layer.forward(apple, apple_num)
price = mul_tax_layer.forward(apple_price, tax)
print(price)

# ---------------------------买苹果的例子，反向传播------------------------------------------------------
"""反向传播的过程计算时，只需要跟正向传播反过来即可"""
dprice = 1
dapple_price, dtax = mul_tax_layer.backward(dprice)
dapple, dapple_num = mul_apple_layer.backward(dapple_price)
print(dapple, dapple_num, dtax)