Fix sampling when more than 2 labels (#1449) #5905

Job	Run time
Formatting	13s
Save-PR-Number	0s
GPU-HIP	15m 20s
GPU-SYCL	30m 13s
CPU (ubuntu-22.04, Release, NoOpenMP)	13m 26s
Docker	14m 44s
CPU (ubuntu-22.04, Debug, OpenMP)	11m 10s
GPU-CUDA	26m 48s
CPU (ubuntu-22.04, Debug, NoOpenMP)	10m 54s
Lint-clang-tidy	33m 26s
CPU (macos-latest, Release, NoOpenMP)	7m 8s
Lint-codeql	28m 18s
Lint-codespell	5s
	3h 11m 45s

Provide feedback