Robuta

https://www.tensorflow.org/api_docs/python/tf/experimental/BatchableExtensionType?authuser=0 tf.experimental.BatchableExtensionType | TensorFlow v2.16.1 An ExtensionType that can be batched and unbatched. tfexperimentaltensorflow