pytorch版本實(shí)現(xiàn)的t-SNE,可以支持cuda加速,根據(jù)作者的python版本修改獲得
MNIST數(shù)據(jù)集在pytorch版本下的結(jié)果

pytorch
對(duì)比原來(lái)python版本實(shí)現(xiàn)的結(jié)果

python
具體源碼,見(jiàn)github
https://github.com/mxl1990/tsne-pytorch
pytorch版本實(shí)現(xiàn)的t-SNE,可以支持cuda加速,根據(jù)作者的python版本修改獲得
MNIST數(shù)據(jù)集在pytorch版本下的結(jié)果
對(duì)比原來(lái)python版本實(shí)現(xiàn)的結(jié)果
具體源碼,見(jiàn)github
https://github.com/mxl1990/tsne-pytorch