tf.nn.softmax.md 637 字节
Newer Older
1
## tf.nn.softmax
J
Jason 已提交
2

3
### [tf.nn.softmax](https://www.tensorflow.org/api_docs/python/tf/nn/softmax)
J
Jason 已提交
4 5 6 7 8 9 10 11 12 13 14

``` python
tf.nn.softmax(
    logits,
    axis=None,
    name=None,
    dim=None
)
```


J
Jason 已提交
15
### [paddle.fluid.layers.softmax](http://paddlepaddle.org/documentation/docs/zh/1.3/api_cn/layers_cn.html#paddle.fluid.layers.softmax)
J
Jason 已提交
16 17 18 19 20 21 22 23 24 25 26
``` python
paddle.fluid.layers.softmax(
    input, 
    use_cudnn=True, 
    name=None
)
```
### 功能差异

#### 计算方式
TensorFlow: 通过`axis`参数,在指定维度上进行计算,如若`axis``None`,则表示在最后一维上进行计算;  
J
Jason 已提交
27
PaddlePaddle:仅在最后一维上进行计算。