如何在CUDA中为Transformer编写一个PyTorch自定义层
https://cloud.tencent.com/developer/article/1419577
Pytorch拓展进阶(一):Pytorch结合C以及Cuda语言
https://oldpan.me/archives/pytorch-combine-c-and-cuda
https://cloud.tencent.com/developer/article/1419577
https://oldpan.me/archives/pytorch-combine-c-and-cuda