Add support for pytorch.tensor

I'm interested in adding support for pytorch tensor, such that users can use it directly without constantly convert the data type.

The coding is trivial, I'm already taking hands on it. but this may introduce optional dependencies of pytorch. I'm not sure how to edit the setup.py file.

Please let me know what you think of this feature and if you can help me with the dependency problem