fix a little bug in shuffle

[badboynt1]

What type of PR is this?

• [ ] API-change
• [x] BUG
• [ ] Improvement
• [ ] Documentation
• [ ] Feature
• [ ] Test and CI
• [ ] Code Refactoring

Which issue(s) this PR fixes:

issue #9747

What this PR does / why we need it:

if data distribution is not uniformed, should use shuffle range to avoid shuffle imbalance