k_map_fn function

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