【Python脚本】绘制常见激活函数代码

import matplotlib.pyplot as plt
import numpy as np
import math

def elu(x,alpha=1):
    a = x[x>0]
    b = alpha*(math.e**(x[x<0])-1)
    result=np.concatenate((b,a),axis=0)
    return result

def relu(x):
    if x >= 0:
        return x
    else :
        return np.exp(x)-1
    #return np.maximum(0,x)

x = np.arange(-10, 10, 0.1)
y=elu(x)
plt.title("ELU")
plt.plot(x, y)
plt.show()