Skip to main content

Browse the archive

https://github.com/tensorly/tensorly
24 November 2021, 14:44:50 UTC
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 
Add where() to Pytorch backend
 
Pytorch does not seem to include the `where()` function so we implement our own.
1 parent 205556a
Raw File
Tip revision: f8120a7ca395d98164b2d8a092e4ef4c5bc48b7f authored by Chris Swierczewski on 01 February 2018, 18:38:32 UTC
Add where() to Pytorch backend
Tip revision: f8120a7
Makefile 
# Automate testing etc
BACKEND?='mxnet'

all: install test

install:
	pip install -e .

test:
	TENSORLY_BACKEND=$(BACKEND) pytest -v tensorly

test-all:
	TENSORLY_BACKEND='mxnet' pytest -v tensorly
	TENSORLY_BACKEND='numpy' pytest -v tensorly
	TENSORLY_BACKEND='pytorch' pytest -v tensorly

test-coverage:
	TENSORLY_BACKEND=$(BACKEND) pytest -v --cov tensorly tensorly
back to top