UserWarning:Implicit dimension choice for softmax has been deprecated.原理及解决

前言：

博主最近在搭建模型的时候，突然爆出警告，可能是新版本的函数的问题。

一、softmax函数

softmax用于多分类过程中，它将多个神经元的输出，映射到（0,1）区间内，可以看成概率来理解，从而来进行多分类！

假设我们有一个数组，V，Vi表示V中的第i个元素，那么这个元素的softmax值就是

更形象的如下图表示：

softmax直白来说就是将原来输出是3,1,-3通过softmax函数一作用，就映射成为(0,1)的值，而这些值的累和为1（满足概率的性质），那么我们就可以将它理解成概率，在最后选取输出结点的时候，我们就可以选取概率最大（也就是值对应最大的）结点，作为我们的预测目标！

二、问题解决

2.1 softmax帮助文档：

Args:
    dim (int): A dimension along which Softmax will be compu