我正在开发一个新的自定义NN应用程序,其中我需要模型来输出一维矢量。输出中的所有值都必须介于零和1之间,我需要向量具有单位范数(即所有值的总和= 1)。
我对NN有一定的经验,但还是一个初学者。据我了解,这与限制输出层的激活有关。现在,我的输出层是一个具有10个节点的密集层。使用sigmoid激活函数,输出值将是一个10元素向量,其中所有值都在零和1之间。这很好,但我还想强制所有10个值的总和为1。这可能吗?
更新 就像@MatiasValdenegro指出的那样,答案是使用softmax激活。尽管这不是通用解决方案,但在我看来,这正是我一直在寻找的解决方案。
来自https://medium.com/data-science-bootcamp/understand-the-softmax-function-in-minutes-f3a59641e86d
谢谢!