tf.keras.backend.map_fn

TensorFlow 1 version View source on GitHub

Map the function fn over the elements elems and return the outputs.

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

Arguments:

  • fn: Callable that will be called upon each element in elems
  • elems: tensor
  • name: A string name for the map node in the graph
  • dtype: Output data type.

Returns:

Tensor with dtype dtype.