tf.compat.v1.py_func

一、作用

包装一个Python函数，将它用在TensorFlow的操作中。

二、语法

tf.compat.v1.py_func(
    func,
    inp,
    Tout,
    stateful=True,
    name=None
)

给定一个Python函数 func，这个函数将numpy数据作为参数，然后返回numpy数组作为输出，包装这个函数用在TensorFlow图的操作中。

三、参数

Args:

• func：一个Python函数
• inp：一个 Tensor对象的列表
• Tout：一个tensorflow数据类型的列表或者元组，或者一个单一的tensorflow数据类型
• stateful：布尔型，（可选）
• name：这次操作的名称（可选）

Return:

返回一个有func计算的Tensor列表，或者单一张量。

四、例子

def my_func(x):
	return np.sinh(x)

input = tf.compat.v1.placeholder(tf.float32)
y = tf.compat.v1.py_func(my_func, [input], tf.float32)