map применяет функцию к каждому элементу; если функция — TF ops/tf.function, выполняется в графе; чистый Python может тормозить или требовать py_function.
Разбор
- Предпочитайте TF ops в map.
py_function удобен, но медленнее и хуже сериализуется.num_parallel_calls=AUTOTUNE.- Ошибки shape часто всплывают после batch.
Итог
map должен быть tf-native для скорости пайплайна.