【Hackathon 9th No.37】add test_top_k_renorm_probs by Echo-Nie · Pull Request #3755 · PaddlePaddle/FastDeploy (original) (raw)

=== Running check ===
Input probs shape: (3, 6)
Input top_k: [1 2 4]
W0914 10:01:09.158907  2341 gpu_resources.cc:114] Please NOTE: device: 0, GPU Compute Capability: 8.0, Driver API Version: 13.0, Runtime API Version: 12.8
Output shape: (3, 6)
Output probs:
 [[0.         1.         0.         0.         0.         0.        ]
 [0.         0.47175145 0.         0.         0.         0.5282486 ]
 [0.44425836 0.11332123 0.         0.         0.16236813 0.28005224]]
Batch 0: sum=1.000000
Batch 0: non-top-k indices [0, 2, 3, 4, 5] should be zero
Batch 1: sum=1.000000
Batch 1: non-top-k indices [0, 2, 3, 4] should be zero
Batch 2: sum=1.000000
Batch 2: non-top-k indices [2, 3] should be zero
.
=== Running check ===
Input probs shape: (2, 5)
Input top_k: [2 3]
Output shape: (2, 5)
Output probs:
 [[0.         0.5649906  0.4350094  0.         0.        ]
 [0.         0.         0.3981753  0.2763285  0.32549617]]
Batch 0: sum=1.000000
Batch 0: non-top-k indices [0, 3, 4] should be zero
Batch 1: sum=1.000000
Batch 1: non-top-k indices [0, 1] should be zero
.
=== Running check ===
Input probs shape: (1, 5)
Input top_k: [2]
Output shape: (1, 5)
Output probs:
 [[0.        0.5649906 0.4350094 0.        0.       ]]
Batch 0: sum=1.000000
Batch 0: non-top-k indices [0, 3, 4] should be zero
.
=== Running check ===
Input probs shape: (1, 4)
Input top_k: [1]
Output shape: (1, 4)
Output probs:
 [[0. 0. 1. 0.]]
Batch 0: sum=1.000000
Batch 0: non-top-k indices [0, 1, 3] should be zero

Edge case top_k=vocab_size
Input probs:
 [[0.1 0.3 0.4 0.2]]
Output probs:
 [[0.1 0.3 0.4 0.2]]
.
----------------------------------------------------------------------
Ran 4 tests in 0.197s

OK