tf.py_func的使用方法

使用方法"/>

tf.py_func的使用方法

tf.py_func

• • 参数：
  • **返回值**

最近在看Faster RCNN源码时，发现了tf.py_func这样一个函数，之前从来没见过，所以就去查了一下关于这个函数的一些资料。

这个函数的主要作用是包装一个Python的普通函数，使这个函数可以作为tensorflow的一个节点op去使用，扩展了Tensorflow的灵活性。
这是官网的解释：Wraps a python function into a TensorFlow op that executes it eagerly.

py_func(func,inp,Tout,stateful=True,name=None
) 

**

参数：

**：
func: 一个 Python 函数, 它接受 NumPy 数组作为输入和输出，并且数组的类型和大小必须和输入和输出用来衔接的 Tensor 大小和数据类型相匹配.
inp: 输入的 Tensor 列表.
Tout: 输出 Tensor 数据类型的列表或元组.
stateful: 状态，布尔值.name: 节点 OP 的名称

返回值

由张量或单个张量组成的函数计算的列表;如果func返回None，则为空列表。