大佬教程收集整理的这篇文章主要介绍了在不离开 GPU 的情况下将 PyTorch GPU 张量转换为 Tensorflow 1.x 张量,大佬教程大佬觉得挺不错的,现在分享给大家,也给大家做个参考。
我在 GPU 中有 PyTorch 张量,我想在不离开 GPU 的情况下将其转换为 Tensorflow 1.x 张量。
我不在乎来回传递渐变。我只是想让张量留在 GPU 中。我不在乎我的 PyTorch 张量是否被复制到 TF 张量或就地修改。
我知道我可以convert to numpy on the CPU并重新加载到 GPU 上。但是,为了速度,我必须继续使用 GPU。
暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!
如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。
小编邮箱:dio#foxmail.com (将#修改为@)
以上是大佬教程为你收集整理的在不离开 GPU 的情况下将 PyTorch GPU 张量转换为 Tensorflow 1.x 张量全部内容,希望文章能够帮你解决在不离开 GPU 的情况下将 PyTorch GPU 张量转换为 Tensorflow 1.x 张量所遇到的程序开发问题。
如果觉得大佬教程网站内容还不错,欢迎将大佬教程推荐给程序员好友。
本图文内容来源于网友网络收集整理提供,作为学习参考使用,版权属于原作者。
如您有任何意见或建议可联系处理。小编QQ:384754419,请注明来意。