tf.raw_ops.WindowDataset
Объединяет (вложенные) элементы входных данных в набор (вложенных) окон.
tf.raw_ops.WindowDataset(
input_dataset, size, shift, stride, drop_remainder, output_types, output_shapes,
name=None
)
«Окно» — это конечный набор плоских элементов размером size (или, возможно, меньше, если для заполнения окна недостаточно элементов входных данных и drop_remainder имеет значение false).
Аргумент shift определяет количество элементов входных данных, на которое окно сдвигается на каждой итерации. Первый элемент в k-ом окне будет элементом
1 + (k-1) * shift
набора входных данных. В частности, первый элемент первого окна всегда будет первым элементом входного набора данных.
Если параметр stride больше 1, каждое окно будет пропускать (stride - 1) элементов входных данных между каждым элементом, который отображается в окне. Результирующие окна по-прежнему будут содержать size элементов независимо от значения stride.
Аргумент stride определяет шаг элементов входных данных, а аргумент shift определяет смещение окна.
Например, пусть {...} представляет собой набор данных:
-
tf.data.Dataset.range(7).window(2)производит{ {0, 1}, {2, 3}, {4, 5}, {6} } -
tf.data.Dataset.range(7).window(3, 2, 1, True)производит{ {0, 1, 2}, {2, 3, 4}, {4, 5, 6} } -
tf.data.Dataset.range(7).window(3, 1, 2, True)производит{ {0, 2, 4}, {1, 3, 5}, {2, 4, 6} }
Обратите внимание, что когда преобразование window применяется к набору данных вложенных элементов, оно производит набор данных вложенных окон.
Например:
-
tf.data.Dataset.from_tensor_slices((range(4), range(4))).window(2)производит{({0, 1}, {0, 1}), ({2, 3}, {2, 3})} -
tf.data.Dataset.from_tensor_slices({"a": range(4)}).window(2)производит{ {"a": {0, 1} }, {"a": {2, 3} } }
| Аргументы | |
|---|---|
input_dataset | Набор данных типа Tensor. |
size | Целочисленное скалярное значение типа int64. Представляет количество элементов набора входных данных, объединяемых в окно. Должно быть положительным. |
shift | Целочисленное скалярное значение типа int64. Представляет количество элементов входных данных, на которое окно сдвигается на каждой итерации. По умолчанию size. Должно быть положительным. |
stride | Целочисленное скалярное значение типа int64. Представляет шаг элементов входных данных в скользящем окне. Должно быть положительным. Значение по умолчанию 1 означает «сохранять каждый элемент входных данных». |
drop_remainder | Логическое скалярное значение типа bool. Указывает, следует ли отбрасывать последнее окно, если его размер меньше window_size. |
output_types | Список tf.DTypes, длина которого равна >= 1. |
output_shapes | Список форм (каждая — tf.TensorShape или список ints), длина которого равна >= 1. |
name | Имя операции (необязательно). |
| Возвращаемое значение | |
|---|---|
Набор данных типа variant. |
© 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/raw_ops/WindowDataset