给我写一个绘制softmax函数和SiLU函数图像的代码
时间: 2024-10-05 11:01:42 浏览: 61
为了绘制softmax函数和SiLU(Sigmoid Linear Unit)函数的图像,我们可以使用Python的matplotlib库。这里是一个简单的例子:
```python
import numpy as np
import matplotlib.pyplot as plt
# 定义sigmoid和softmax函数
def sigmoid(x):
return 1 / (1 + np.exp(-x))
def softmax(logits):
exp_logits = np.exp(logits)
return exp_logits / exp_logits.sum(axis=1, keepdims=True)
# 创建数据范围
x = np.linspace(-8, 8, 400)
# 计算softmax和sigmoid在该范围的值
sigmoid_values = sigmoid(x)
softmax_values = softmax(x)
# 绘制两条曲线
plt.figure(figsize=(8, 6))
plt.plot(x, sigmoid_values, label='Sigmoid')
plt.plot(x, softmax_values, label='Softmax')
# 添加图例、标签和标题
plt.title('Comparison of Sigmoid and Softmax Functions')
plt.xlabel('Input Values')
plt.ylabel('Output Value')
plt.legend()
plt.grid(True)
plt.show()
阅读全文