tf.data.experimental.group_by_reducer
Преобразование, которое группирует элементы и выполняет сведение.
tf.data.experimental.group_by_reducer(
key_func, reducer
)
Это преобразование сопоставляет элемент набора данных с ключом с помощью key_func и группирует элементы по ключу. Для обработки каждой группы используется reducer; его init_func используется для инициализации состояния каждой группы при ее создании, reduce_func используется для обновления состояния каждый раз, когда элемент сопоставляется с соответствующей группой, а finalize_func используется для преобразования конечного состояния в выходное значение.
| Аргументы | |
|---|---|
key_func | Функция, сопоставляющая вложенную структуру тензоров (формы и типы которых определяются с помощью self.output_shapes и self.output_types) со скалярным тензором tf.int64. |
reducer | Экземпляр Reducer, содержащий логику сведения с помощью функций init_func, reduce_func и finalize_func. |
| Возвращает | |
|---|---|
Функцию преобразования Dataset, которую можно передать в tf.data.Dataset.apply. |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/data/experimental/group_by_reducer
