Skip to content

Commit 1829c3a

Browse files
committed
Remove check in top_k API. (#13178)
1 parent 89e65a2 commit 1829c3a

File tree

1 file changed

+0
-5
lines changed
  • python/paddle/fluid/layers

1 file changed

+0
-5
lines changed

python/paddle/fluid/layers/nn.py

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -3496,11 +3496,6 @@ def topk(input, k, name=None):
34963496
34973497
top5_values, top5_indices = layers.topk(input, k=5)
34983498
"""
3499-
shape = input.shape
3500-
if k < 1 or k >= shape[-1]:
3501-
raise ValueError("k must be greater than 0 and less than %d." %
3502-
(shape[-1]))
3503-
35043499
helper = LayerHelper("top_k", **locals())
35053500
values = helper.create_tmp_variable(dtype=input.dtype)
35063501
indices = helper.create_tmp_variable(dtype="int64")

0 commit comments

Comments
 (0)