Update tanh_layer.py