tf.data.experimental.group_by_window
| Просмотреть исходный код на GitHub |
Преобразование, которое группирует окна элементов по ключу и уменьшает их.
tf.data.experimental.group_by_window(
key_func, reduce_func, window_size=None, window_size_func=None
)
Это преобразование сопоставляет каждый последовательный элемент в наборе данных с ключом с помощью key_func и группирует элементы по ключу. Затем оно применяет reduce_func к не более чем window_size_func(key) элементам, соответствующим одному и тому же ключу. Все, кроме последнего окна для каждого ключа, будут содержать window_size_func(key) элементов; последнее окно может быть меньше.
Вы можете указать либо постоянное значение window_size, либо размер окна, определяемый ключом, с помощью window_size_func.
| Аргументы | |
|---|---|
key_func | Функция, сопоставляющая вложенную структуру тензоров (с формами и типами, определёнными self.output_shapes и self.output_types), со скалярным тензором tf.int64. |
reduce_func | Функция, сопоставляющая ключ и набор данных из не более чем window_size последовательных элементов, соответствующих этому ключу, с другим набором данных. |
window_size | Скалярный tf.int64 tf.Tensor, представляющий количество последовательных элементов, соответствующих одному и тому же ключу, которые нужно объединить в одну партию, которое будет передано в reduce_func. Взаимоисключающее с window_size_func. |
window_size_func | Функция, сопоставляющая ключ со скалярным tf.int64 tf.Tensor, представляющим количество последовательных элементов, соответствующих одному и тому же ключу, которые нужно объединить в одну партию, которое будет передано в reduce_func. Взаимоисключающее с window_size. |
| Возвращаемое значение | |
|---|---|
Функция преобразования Dataset, которая может быть передана в tf.data.Dataset.apply. |
| Исключения | |
|---|---|
ValueError | если ни один, ни оба из {window_size, window_size_func} не переданы. |
© 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.4/api_docs/python/tf/data/experimental/group_by_window
