The later night swim comes after she finally confirmed her romance with the actor Nat Wolff when they were spotted kissing in Venice last week