基本的例子

tf.py_func(func, inp, Tout) 運算子建立一個 TensorFlow 操作，該操作在張量感測器 inp 上呼叫 Python 函式 func。

請參閱 tf.py_func(func, inp, Tout) 的文件 。

警告 ：tf.py_func() 操作只能在 CPU 上執行。如果你使用的是分散式 TensorFlow，則必須在與客戶端相同的程序中將 tf.py_func() 操作放在 CPU 裝置上。

def func(x):
    return 2*x

x = tf.constant(1.)
res = tf.py_func(func, [x], [tf.float32])
# res is a list of length 1