US soldiers have been training in Latvia by plunging into icy water in a hole in a frozen lake.