tfpyth:将TensorFlow放回PyTorch中再放回TensorFlow中(不同的TensorFlow PyTorch适配器) 源码

苯丙氨酸 将TensorFlow放回PyTorch中,再放回TensorFlow中(使用可区分的TensorFlow PyTorch适配器)。 您是否有一个使用TensorFlow的代码库和一个使用PyTorch的代码库,并且想要训练一个同时使用端到端的模型? 该库无需重写任何一个代码库就可以实现! 它允许您使用简单的函数包装TensorFlow图以使其可以通过PyTorch进行调用(和区分),反之亦然。 唯一的警告是张量必须通过CPU复制并路由,直到TensorFlow支持__cuda_array_interface (请在__cuda_array_interface星)。 安装