tf.data.experimental.group_by_reducer
| Просмотреть исходный код на GitHub |
Преобразование, группирующее элементы и выполняющее редукцию.
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. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.3/api_docs/python/tf/data/experimental/group_by_reducer
