Visual Token Pruning by Attention Mass

Unlock premium problems

This is a Premium problem. Upgrade to unlock the full editor, test cases, and solution.