Incompatibilities with Tensorflow 2

I've recently started using this for a project, and I noticed that I was having issues because of my Tensorflow version.

In the PR #13 you can find the changes I did to update the code to the new Tensorflow version. Unfortunately, I couldn't test the changes thoroughly, especially in the examples.