tf.keras.backend.map_fn

TensorFlow 2 version View source on GitHub

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

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.

Tensor with dtype dtype.