TSNE Pytorch based implementation Only tested on Ubuntu18.04 with CUDA9.1, Pytorch 0.4.0 speed boost test environment CPU: