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
File | Mode | Size |
---|---|---|
backend | ||
datasets | ||
decomposition | ||
metrics | ||
random | ||
regression | ||
tenalg | ||
tests | ||
__init__.py | -rw-r--r-- | 1.6 KB |
base.py | -rw-r--r-- | 5.6 KB |
kruskal_tensor.py | -rw-r--r-- | 2.7 KB |
tucker_tensor.py | -rw-r--r-- | 2.9 KB |
Computing file changes ...