pytorch_extention a simple implementation demo by using pytorch extention pytorch == 0.4.0 python main.py