TensorFlow函数教程：tf.keras.backend.map_fn

tf.keras.backend.map_fn函数

tf.keras.backend.map_fn(
    fn,
    elems,
    name=None,
    dtype=None
)

定义在：tensorflow/python/keras/backend.py。

将函数fn映射到元素elems并返回输出。

参数：

• fn：可调用，将调用elems中的每个元素。
• elems：张量
• name：图中映射节点的字符串名称
• dtype：输出数据类型。

返回：

类型为dtype的张量。