tensorflow - 在不离开 GPU 的情况下将 PyTorch GPU 张量转换为 Tensorflow 1.x 张量

Question

我在 GPU 中有 PyTorch 张量，我想在不离开 GPU 的情况下将其转换为 Tensorflow 1.x 张量。

我不关心来回传递渐变。我只希望张量留在 GPU 中。我不在乎我的 PyTorch 张量是否被复制到 TF 张量或就地修改。

我知道我可以在 CPU 上转换为 numpy并重新加载到 GPU 上。但是，为了速度，我必须留在 GPU 上。