0
点赞
收藏
分享

微信扫一扫

助力汽车半导体产业发展,2025 广州国际新能源汽车功率半导体技术展览会与您相约“羊城”广州

 1 人工神经网络

全连接神经网络

2 激活函数

  • 隐藏层激活函数由人决定
  • 输出层激活函数由解决的任务决定:
    • 二分类:sigmoid
    • 多分类:softmax
    • 回归:不加激活(恒等激活identify)

2.1 sigmoid激活函数

  • x为加权和
  • 小于-6或者大于6,梯度接近于0,会出现梯度消失的问题
  • 即使取值 [-6,6] ,网络超过5层,也会发生梯度消失

import torch
import matplotlib.pyplot as plt
import os

os.environ["KMP_DUPLICATE_LIB_OK"] = "TRUE"
# sigmoid
x = torch.linspace(-15, 15, 1000)
y = torch.sigmoid(x)
plt.plot(x, y)
plt.grid()
plt.show()

x = torch.linspace(-15, 15, 1000, requires_grad=True)
torch.sigmoid(x).sum().backward()
plt.plot(x.detach(), x.grad)
plt.grid()
plt.show()

2.2 tanh激活函数

  • 只在RNN使用

举报

相关推荐

0 条评论