TensorFlow的高阶函数 tf.map_fn()

 

这里有不错的解释：https://blog.youkuaiyun.com/loseinvain/article/details/78815130

50个例子：https://www.programcreek.com/python/example/90420/tensorflow.map_fn

更多解说：https://code.i-harness.com/ja/docs/tensorflow~python/tf/map_fn

 

简单的总结一下：

定义：

tf.map_fn(
    fn,
    elems,
    dtype=None,
    parallel_iterations=10,
    back_prop=True,
    swap_memory=False,
    infer_shape=True,
    name=None
)

把函数当参数传进去，可以直接用lambda。
比较有用的，可以理解为可以把多个fn函数，并行执行。
将参数elems从第一维展开，进行map处理。

 

一个简单的例子

def fun1(a):
    if a.shape == (1,2):
        prin