未验证 提交 c725030d 编写于 作者: S SunAhong1993 提交者: GitHub

Update Sofmax.md

上级 ba053d9e
...@@ -16,8 +16,9 @@ layer { ...@@ -16,8 +16,9 @@ layer {
```python ```python
paddle.fluid.layers.softmax( paddle.fluid.layers.softmax(
input, input,
use_cudnn=True, use_cudnn=False,
name=None name=None,
axis=-1
) )
``` ```
...@@ -25,3 +26,5 @@ paddle.fluid.layers.softmax( ...@@ -25,3 +26,5 @@ paddle.fluid.layers.softmax(
#### 计算机制 #### 计算机制
Caffe:计算softmax之前,对每个样本中的每个值减去该样本中的最大值; Caffe:计算softmax之前,对每个样本中的每个值减去该样本中的最大值;
PaddlePaddle:省略了这一操作直接计算softmax。 PaddlePaddle:省略了这一操作直接计算softmax。
#### 使用机制
PaddlePaddle:通过设置`axis`来确定执行softmax的维度索引。
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册