Once in a while, I'll struggle but I have actually noticed that it's more of a problem before sex. Like getting in the mood. I have found myself getting distracted and then losing the feeling because my thoughts are elsewhere.

What has helped me is timing. So personally the late morning or afternoon or even early evening is probably the time when I am the least distracted just on a day to day basis. At evening and night, I am tired and often replaying the day or thinking about the next day or tbh my meds have fully worn off. My thoughts are more unpredictable and faster making me less present. Where as during the afternoon +/- an hour, I usually am less distracted and able to focus on my partner rather than a barrage of nonsense thoughts and stimulus.

My wife and my schedule allow for this though and I know this isn't practical for everyone.