当前位置：网站首页>激活函数之阶跃函数

激活函数之阶跃函数

2022-04-23 11:36:00 【小嘿黑15斤】

阶跃函数

阶跃函数以 0 为界，输出从 0 切换为 1（或者从 1 切换为 0）。它的值呈阶梯式变化，所以称为阶跃函数。

import matplotlib.pyplot as plt
import numpy as np

def step_function(x):
    y = x>0
    print(y.shape,y.dtype)
    return y.astype(np.int32)

# def step_function(x):
# print(x>0)
# return np.array(x>0,dtype=np.int32)


if __name__ == '__main__':

    x = np.arange(-5,5,0.1)
    y = step_function(x)

    plt.plot(x,y)
    plt.ylim(-0.1,1.1)
    plt.show()