Python 实现神经网络 ReLU 函数测试
以下是一个简单的 Python 示例代码,用于测试 ReLU 函数:
import numpy as np
def relu(x):
return np.maximum(0, x)
# 测试 ReLU 函数
x = np.array([-1, 0, 1, 2, 3])
print(relu(x))
输出结果:
[0 0 1 2 3]
在这个示例中,我们定义了一个名为 'relu' 的函数,该函数接受一个 NumPy 数组作为输入,并返回一个应用 ReLU 函数的 NumPy 数组。我们使用 'np.maximum' 函数来实现 ReLU 函数,该函数返回两个输入值中的最大值。注意,当输入值小于 0 时,'np.maximum' 返回 0,这正是 ReLU 函数的定义。
我们还使用 NumPy 库来创建输入数组 'x' 和计算 ReLU 函数的输出。在测试代码中,我们打印出 ReLU 函数的输出,以确保它按预期工作。
原文地址: https://www.cveoy.top/t/topic/nu0s 著作权归作者所有。请勿转载和采集!