一、作用
包装一个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)