tf.nn.top_k.md 809 字节
Newer Older
J
jiangjiajun 已提交
1 2
## tf.nn.top_k

J
jiangjiajun 已提交
3
### [tf.nn.top_k](https://www.tensorflow.org/versions/r1.13/api_docs/python/tf/nn/top_k)
J
jiangjiajun 已提交
4 5 6 7 8 9 10 11 12
``` python
tf.math.top_k(
    input,
    k=1,
    sorted=True,
    name=None
)
```

J
jiangjiajun 已提交
13
### [paddle.fluid.layers.topk](http://paddlepaddle.org/documentation/docs/zh/1.4/api_cn/layers_cn.html#topk)
J
jiangjiajun 已提交
14 15 16 17 18 19 20 21 22 23 24
``` python
paddle.fluid.layers.topk(
    input, 
    k, 
    name=None
)
```

### 功能差异
#### 参数差异
TensorFlow: 通过设置`sorted`参数,对返回的值与下标设置是否进行降序排序;`k`默认为1。  
J
Jason 已提交
25
PaddlePaddle: 对返回的top-k tensor进行降序排序;`k`没有默认值,必须设置。
J
jiangjiajun 已提交
26 27 28 29 30 31 32 33

### 代码示例
```python
# 输入 tensor t 为[[2,6,3],[3,0,8]]

# 当k=2时,输出 tensor out 为[[6,3], [8,3]],index为[[1,2],[2,0]]
out, index = fluid.layers.topk(t, k=1)

J
jiangjiajun 已提交
34
```