add torch stuff
Merge pull request #1 from timpohladthomas/patch-1

add torch stuff