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的张量。

以上内容是否对您有帮助:
在线笔记
App下载
App下载

扫描二维码

下载编程狮App

公众号
微信公众号

编程狮公众号

意见反馈
返回顶部