Whenever I swim under water my screen goes black and I can't see anything.

Thank you for your report!
We're actually already tracking this issue at MCPE-32799, so we will resolve and link this ticket as a duplicate.

If you would like to add a vote and any extra information to the main ticket it would be appreciated.

If you haven't already, you might like to make use of the search feature to see if the issue has already been mentioned.