基本的例子

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