https://github.com/tensorly/tensorly
Revision f8120a7ca395d98164b2d8a092e4ef4c5bc48b7f authored by Chris Swierczewski on 01 February 2018, 18:38:32 UTC, committed by Jean Kossaifi on 03 February 2018, 18:53:37 UTC
Pytorch does not seem to include the `where()` function so we implement our own.
1 parent 205556a
Tip revision: f8120a7ca395d98164b2d8a092e4ef4c5bc48b7f authored by Chris Swierczewski on 01 February 2018, 18:38:32 UTC
Add where() to Pytorch backend
Add where() to Pytorch backend
Tip revision: f8120a7
setup.py
try:
from setuptools import setup, find_packages
except ImportError:
from distutils.core import setup, find_packages
import tensorly
version = tensorly.__version__
def readme():
with open('README.rst') as f:
return f.read()
config = {
'name': 'tensorly',
'packages': find_packages(exclude=['doc']),
'description': 'Tensor learning in Python.',
'long_description': readme(),
'author': 'Jean Kossaifi',
'author_email': 'jean.kossaifi@gmail.com',
'version': version,
'url': 'https://github.com/tensorly/tensorly',
'download_url': 'https://github.com/tensorly/tensorly/tarball/' + version,
'install_requires': ['numpy', 'scipy'],
'license': 'Modified BSD',
'scripts': [],
'classifiers': [
'Topic :: Scientific/Engineering',
'License :: OSI Approved :: BSD License',
'Programming Language :: Python :: 3'
],
}
setup(**config)
Computing file changes ...