Hello! Unfortunately you cannot use that expression to tell someone to swim in the ocean. “Take a bath” is something we say when we want to tell the person to shower or bathe in a bathtub. Instead, you would say “go swim” or “would you like to swim in the ocean?”. Hope this helps!