YaoooLiang / pytorch_extention

pytorch结合C以及CUDA

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

pytorch_extention

a simple implementation demo by using pytorch extention

pytorch == 0.4.0

python main.py

About

pytorch结合C以及CUDA


Languages

Language:Python 71.5%Language:C 21.1%Language:C++ 7.4%