Merge branch 'master' of github.com:parrt/tensor-sensor into master

Author: parrt

README.md

@@ -39,7 +39,10 @@ Cause: @ on tensor operand W w/shape [764, 100] and operand X.T w/shape [764, 20
 ## Install
 
 ```
-pip install tensor-sensor
+pip install tensor-sensor # This will only install the library for you
+pip install tensor-sensor[torch] # install pytorch related dependency
+pip install tensor-sensor[tensorflow] # install tensorflow related dependency
+pip install tensor-sensor[all] # install both tensorflow and pytorch
 ```
 
 which gives you module `tsensor`. I developed and tested with the following versions
@@ -94,4 +97,4 @@ $ pip install .
 
 * can i call pyviz in debugger?
 * try on real examples
-* `dict(W=[3,0,1,2], b=[1,0])` that would indicate (300, 30, 60, 3) would best be displayed as (30,60,3, 300) and b would be  first dimension last and last dimension first
+* `dict(W=[3,0,1,2], b=[1,0])` that would indicate (300, 30, 60, 3) would best be displayed as (30,60,3, 300) and b would be  first dimension last and last dimension first

contributing.md

@@ -0,0 +1,3 @@
+Thank you for considering a contribution to TensorSensor.  I welcome updates to the examples, README, or the source code itself.
+
+Before you spend a lot of time creating a pull request (PR) for a new feature or a biggish change to the software, please contact me by email [email protected]  to find out if it's something I'm interested in.  One of the hardest things to do, as I've learned over the last 30 years pushing out open-source software, is to maintain project focus and conceptual integrity.

setup.py

@@ -23,6 +23,10 @@
 """
 from setuptools import setup
 
+tensorflow_requires = ['tensorflow']
+torch_requires = ['torch']
+all_requires = tensorflow_requires + torch_requires
+
 exec(open('tsensor/version.py').read())
 setup(
     name='tensor-sensor',
@@ -33,7 +37,11 @@
     author='Terence Parr',
     author_email='[email protected]',
     python_requires='>=3.6',
-    install_requires=['graphviz>=0.14.1','numpy','torch','tensorflow', 'IPython', 'matplotlib'],
+    install_requires=['graphviz>=0.14.1','numpy','IPython', 'matplotlib'],
+    extras_require = {'all': all_requires,
+                      'torch': torch_requires,
+                      'tensorflow': tensorflow_requires
+                     },
     description='The goal of this library is to generate more helpful exception messages for numpy/pytorch tensor algebra expressions.',
 #    keywords='visualization data structures',
     classifiers=['License :: OSI Approved :: MIT License',