topk

    • 例1：
    1. import paddle.fluid as fluid
    3. input = layers.data(name="input", shape=[13, 11], dtype='float32')
    4. top5_values, top5_indices = layers.topk(input, k=5) #top5_values.shape=[13, 5]，top5_indices.shape=[13, 5]
    5. # 1D Tensor
    7. top5_values, top5_indices = layers.topk(input1, k=5) #top5_values.shape=[5]，top5_indices.shape=[5]
    9. # k=Variable
    10. input2 = layers.data(name="input2", shape=[13, 11], dtype='float32')