Fix flakyness in tests related to torch.unique with dim=None
#4178
| Job | Run time |
|---|---|
| 11s | |
| 20m 54s | |
| 25m 34s | |
| 20m 47s | |
| 28m 40s | |
| 28m 22s | |
| 28m 16s | |
| 28m 25s | |
| 30m 47s | |
| 31m 7s | |
| 20m 5s | |
| 29m 4s | |
| 29m 41s | |
| 26m 48s | |
| 24m 14s | |
| 38m 24s | |
| 17m 36s | |
| 17m 10s | |
| 20m 34s | |
| 26m 42s | |
| 26m 58s | |
| 1s | |
| 8h 40m 20s |