TensorFlow函數(shù)教程：tf.keras.backend.map_fn

tf.keras.backend.map_fn函數(shù)

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

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

將函數(shù)fn映射到元素elems并返回輸出。

參數(shù)：

• fn：可調(diào)用，將調(diào)用elems中的每個(gè)元素。
• elems：張量
• name：圖中映射節(jié)點(diǎn)的字符串名稱
• dtype：輸出數(shù)據(jù)類型。

返回：

類型為dtype的張量。