Threepio Python客户端

Threepio是一个用于在PyTorch、TensorFlow和TensorFlow.js之间翻译命令的多语言库。本节代码库是为Python客户端编写的。

安装

使用包管理器pip安装threepio。

pip install openmined.threepio

用法

import tensorflow as tf
from pythreepio import Threepio, Command

# Create a threepio translating from pytorch -> tensorflow python
threepio = Threepio("torch", "tf", tf)

# Create your command to translate
args = [
    tf.constant([[1, 0], [0, 1]]),
    tf.constant([[0, 1], [1, 0]])
]
kwargs = {}
cmd = Command("add", args, kwargs)

# Translate the command
translated_cmd = threepio.translate(cmd, lookup_command=True)

...

# When you're ready, execute the translation
translated_cmd.execute_routine() # -> [[1, 1], [1, 1]]

命令支持

目前对命令的支持有限。官方支持的命令列在我们测试中。