这里有不错的解释:https://blog.csdn.net/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处理。
一个简单的例子