tf.compat.v1.map_fn

Transforms elems b