1. Learn
    2. /
  2. 课程
    3. /
  3. 使用 PyTorch 的深度学习入门
Connected

道练习

Sigmoid 与 Softmax 函数

sigmoid 与 softmax 是深度学习中关键的激活函数，常作为神经网络的最后一步。

  • Sigmoid 用于二分类任务；
  • Softmax 用于多分类任务。

给定来自网络的预激活输出张量，请应用合适的激活函数以得到最终输出。

torch.nn 已作为 nn 导入。

说明 1/2

undefined XP
  • 1
    • 创建一个 sigmoid 函数并将其应用于 input_tensor，为二分类任务生成概率。
  • 2
    • 创建一个 softmax 函数并将其应用于 input_tensor，为多分类任务生成概率。