Add PyTorch test

[berleon]

add similar tests as for tensorflow for pytorch. It would also be nice to run them on travis.

• [x] pytorch.py
• [x] setup pytorch on travis