The water is warm, but he goes in when its colder also. I don't think he realizes what he's doing, he just wants the stick or the ball.
This is only the 3rd time he has been actually swimming. Before he would go out till his but would start to float and then turn back around. Now he is getting braver.