diff --git a/caffe2fluid/doc/Sofmax.md b/caffe2fluid/doc/Sofmax.md index e9cca71b90907ee1a86f8ed2925c6c655887f881..e7da6236db68d05985e45d074dc2efba24a7ddf8 100644 --- a/caffe2fluid/doc/Sofmax.md +++ b/caffe2fluid/doc/Sofmax.md @@ -16,8 +16,9 @@ layer { ```python paddle.fluid.layers.softmax( input, - use_cudnn=True, - name=None + use_cudnn=False, + name=None, + axis=-1 ) ``` @@ -25,3 +26,5 @@ paddle.fluid.layers.softmax( #### 计算机制 Caffe:计算softmax之前,对每个样本中的每个值减去该样本中的最大值; PaddlePaddle:省略了这一操作直接计算softmax。 +#### 使用机制 +PaddlePaddle:通过设置`axis`来确定执行softmax的维度索引。