Python学习-softmax+crossEntropyLoss

目录

简介

公式

使用

代码

参考


简介

公式

softmax

  • 函数 Softmax(x) 也是一个 non-linearity, 但它的特殊之处在于它通常是网络中一次操作. 这是因为它接受了一个实数向量并返回一个概率分布.其定义如下. 定义 x 是一个实数的向量(正数或负数都无所谓, 没有限制). 然后, 第i个 Softmax(x) 的组成是

输出是一个概率分布:

你可能感兴趣的:(pyg)