Skip to content

Commit 1ca241c

Browse files
authored
Merge pull request #13228 from qingqing01/top_k_api
Remoe check in top_k API
2 parents ed11f49 + a5f0376 commit 1ca241c

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
@@ -3542,11 +3542,6 @@ def topk(input, k, name=None):
35423542
35433543
top5_values, top5_indices = layers.topk(input, k=5)
35443544
"""
3545-
shape = input.shape
3546-
if k < 1 or k >= shape[-1]:
3547-
raise ValueError("k must be greater than 0 and less than %d." %
3548-
(shape[-1]))
3549-
35503545
helper = LayerHelper("top_k", **locals())
35513546
values = helper.create_tmp_variable(dtype=input.dtype)
35523547
indices = helper.create_tmp_variable(dtype="int64")

0 commit comments

Comments
 (0)