PyTorch 自定义 CUDA 算子编写、绑定与调用示例教程

如何在PyTorch中进行数据交换： 1. 用C++ API保存的数据，如何用Python读取。 2. 用Python API保存的数据，如何用C++读取