Remove check in top_k API. (#13178)

7 years ago · bab341213e
parent cdd14f17f1
commit bab341213e
1 changed files with 0 additions and 5 deletions
--- a/python/paddle/fluid/layers/nn.py
+++ b/python/paddle/fluid/layers/nn.py
@ -3546,11 +3546,6 @@ def topk(input, k, name=None):

            top5_values, top5_indices = layers.topk(input, k=5)
    """
-    shape = input.shape
-    if k < 1 or k >= shape[-1]:
-        raise ValueError("k must be greater than 0 and less than %d." %
-                         (shape[-1]))
-
    helper = LayerHelper("top_k", **locals())
    values = helper.create_tmp_variable(dtype=input.dtype)
    indices = helper.create_tmp_variable(dtype="int64")