When you try to learn how to divert your mind, you had better start practising with two different games like poker and chess. Think of these two or any two games simultaneously while focusing more on one particular game. This way you can develop this skill.

Total pull back from one is not a good idea and not a good practice as this is not going to be good in sex.